PyTorch pode acelerar os aplicativos AI/ML usando uma GPU NVIDIA por meio da biblioteca NVIDIA CUDA nativamente, assim como o TensorFlow.
Neste artigo, mostraremos como instalar o PyTorch com suporte para aceleração NVIDIA GPU/CUDA no Debian 12 “Bookworm”.
Tópico de Conteúdo:
- Instalando os drivers de GPU NVIDIA no Debian 12
- Instalando NVIDIA CUDA no Debian 12
- Instalando Python 3 PIP e Python 3 Virtual Environment (venv) no Debian 12
- Criando um ambiente virtual Python 3 para PyTorch
- Atualizando Python 3 PIP para a versão mais recente no ambiente virtual Python 3 PyTorch
- Instalando PyTorch com suporte de aceleração NVIDIA GPU/CUDA no Debian 12
- Ativando o ambiente virtual PyTorch Python 3
- Acessando PyTorch e verificando se a aceleração NVIDIA GPU/CUDA está disponível
- Conclusão
Instalando os drivers de GPU NVIDIA no Debian 12
Para que a aceleração PyTorch NVIDIA GPU/CUDA funcione, você deve instale os drivers de GPU NVIDIA no Debian 12 . Se você precisar de ajuda para instalar os drivers de GPU NVIDIA em seu sistema Debian 12, leia este artigo .
Instalando NVIDIA CUDA no Debian 12
Para que a aceleração PyTorch NVIDIA GPU/CUDA funcione no Debian 12, você deve instale NVIDIA CUDA no Debian 12 . Se você precisar de ajuda para instalar NVIDIA CUDA em seu sistema Debian 12, leia este artigo .
Instalando Python 3 PIP e Python 3 Virtual Environment (venv) no Debian 12
Para instalar o PyTorch no Debian 12, você precisa ter o Python 3 PIP e o ambiente virtual Python (venv) instalados.
Primeiro, atualize o cache do repositório de pacotes APT com o seguinte comando:
$ sudo atualização adequada 
Para instalar o Python 3 PIP e o ambiente virtual Python 3 (venv), execute o seguinte comando:
$ sudo apto instalar python3-pip python3-venv python3-dev Para confirmar a instalação, pressione “Y” e depois pressione
Python 3 PIP e Python 3 venv estão sendo instalados. Demora um pouco para ser concluído.
Neste ponto, Python 3 PIP e Python 3 venv devem estar instalados.
Criando um ambiente virtual Python 3 para PyTorch
A prática padrão para instalar as bibliotecas Python no Debian 12 é instalá-las em um ambiente virtual Python para que não interfiram nos pacotes/bibliotecas Python do sistema.
Para criar um novo ambiente virtual Python 3 para PyTorch no diretório “/opt/pytorch”, execute o seguinte comando:
$ sudo python3 -m venv / optar / lanternaAtualizando Python 3 PIP para a versão mais recente no ambiente virtual Python 3 PyTorch
Para atualizar o Python 3 PIP para a versão mais recente no ambiente virtual Python 3 “/opt/pytorch”, execute o seguinte comando:
$ sudo / optar / lanterna / caixa / pip3 instalar --atualizar pip 
Instalando PyTorch com suporte de aceleração NVIDIA GPU/CUDA no Debian 12
Para que a aceleração PyTorch NVIDIA GPU/CUDA funcione, você deve instalar a versão correta do PyTorch que suporta a versão do driver NVIDIA CUDA que você instalou em seu sistema Debian 12. No momento em que este artigo foi escrito, PyTorch suportava o driver NVIDIA CUDA versões 11.8 e 12.1. Para obter informações atualizadas sobre as versões do driver NVIDIA CUDA suportadas pelo PyTorch, verifique o site oficial do PyTorch .
Para verificar a versão do driver NVIDIA CUDA que você instalou em seu sistema Debian 12, execute o seguinte comando. Como você pode ver, temos NVIDIA CUDA versão 11.8 instalado em nosso sistema Debian 12.
$ nvcc --versão 
Para instalar o PyTorch com suporte NVIDIA CUDA 11.8 no ambiente virtual PyTorch Python 3, execute o seguinte comando:
$ sudo / optar / lanterna / caixa / pip3 instalar tocha tochavisão tochaáudio --index-url https: // baixar.pytorch.org / o que / com 118Para instalar o PyTorch com suporte NVIDIA CUDA 12.1 no ambiente virtual PyTorch Python 3, execute o seguinte comando:
$ sudo / optar / lanterna / caixa / pip3 instalar tocha tochavisão tochaáudioPyTorch está sendo instalado no ambiente virtual PyTorch Python 3. Demora um pouco para ser concluído.
Neste ponto, o PyTorch deve estar instalado no ambiente virtual PyTorch Python 3
Ativando o ambiente virtual PyTorch Python 3
Para ativar o ambiente virtual PyTorch Python “/opt/pytorch”, execute o seguinte comando:
$ . / optar / lanterna / caixa / ativarO ambiente virtual PyTorch Python 3 deve ser ativado.
Acessando PyTorch e verificando se a aceleração NVIDIA GPU/CUDA está disponível
Para abrir o shell interativo do Python 3, execute o seguinte comando:
$ python3O shell interativo do Python 3 deve ser aberto.
Primeiro, importe o PyTorch com a seguinte linha de código:
$ importar tocha 
Para verificar a versão do PyTorch que você instalou, execute a seguinte linha de código. Como você pode ver, estamos executando o PyTorch 2.1.0 com suporte para aceleração NVIDIA CUDA 11.8 (cu118).
$ tocha.__versão__ 
Para verificar se o PyTorch é capaz de usar sua GPU NVIDIA para aceleração NVIDIA CUDA, você também pode executar a seguinte linha de código. Se o suporte NVIDIA CUDA estiver disponível, “True” será impresso.
$ tocha.cuda.is_available ( )Se você tiver várias GPUs instaladas em seu computador, poderá verificar o número de GPUs que o PyTorch pode usar com a seguinte linha de código. Como você pode ver, temos a GPU NVIDIA (RTX 4070) instalada em nosso sistema Debian 12.
$ tocha.cuda.device_count ( )Para sair do shell interativo do Python, execute a seguinte linha de código:
$ desistir ( )Conclusão
Neste artigo, mostramos como instalar o ambiente virtual Python 3 PIP e Python 3 (venv) no Debian 12. Também mostramos como criar um ambiente virtual Python 3 para PyTorch no Debian 12 e como instalar PyTorch com NVIDIA CUDA Suporte para aceleração 11.8 e 12.1 no Debian 12 também. Por fim, mostramos como ativar o ambiente virtual PyTorch Python e acessar o PyTorch no Debian 12.