Como Verificar a Versão do CUDA no Ubuntu [Guia de Comando Fácil]

目次

1. Introdução

CUDA (Compute Unified Device Architecture) é uma plataforma de computação paralela desenvolvida pela NVIDIA que utiliza GPUs. É amplamente usada para aprendizado de máquina, deep learning, renderização 3D e várias outras tarefas computacionais.

Ao usar CUDA em um ambiente Ubuntu, é crucial verificar a versão do CUDA pelos seguintes motivos:

Compatibilidade do Driver

O CUDA requer uma versão específica do driver NVIDIA para funcionar corretamente. Se as versões forem incompatíveis, o CUDA pode não operar adequadamente.

Compatibilidade de Bibliotecas

Bibliotecas como TensorFlow e PyTorch exigem versões específicas do CUDA e do cuDNN. É essencial garantir que você tenha instalado a versão apropriada.

Evitar Confusão no Sistema

Se múltiplas versões do CUDA estiverem instaladas no sistema, é necessário identificar qual versão está ativa e alternar entre elas conforme necessário.

Neste artigo, forneceremos uma explicação clara de como verificar a versão do CUDA no Ubuntu.

2. Como Verificar a Versão do CUDA no Ubuntu

Em um ambiente Ubuntu, você pode checar a versão do CUDA usando os seguintes métodos:

Método 1: Verificar com o Comando nvidia-smi (Método Mais Fácil)

O driver NVIDIA inclui uma ferramenta chamada nvidia-smi (NVIDIA System Management Interface) que permite verificar o status da sua GPU.

Comando de Execução

nvidia-smi

Exemplo de Saída

+-----------------------------------------------------------------------------+
| NVIDIA-SMI 530.41.03    Driver Version: 530.41.03    CUDA Version: 12.1     |
+-----------------------------------------------------------------------------+

Pontos Principais

  • O CUDA Version: 12.1 exibido aqui representa a versão máxima do CUDA suportada pelo driver NVIDIA.
  • Isso pode não coincidir sempre com a versão do toolkit CUDA instalada, portanto é recomendado verificar usando métodos adicionais.

Método 2: Verificar com o Comando nvcc -V (Para Desenvolvedores)

Se o CUDA estiver instalado corretamente, você pode checar a versão do nvcc (o compilador CUDA).

Comando de Execução

nvcc -V

Exemplo de Saída

nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2023 NVIDIA Corporation
Built on Sun_Jul_30_19:09:40_PDT_2023
Cuda compilation tools, release 12.1, V12.1.105

Pontos Principais

  • A parte que diz release 12.1, V12.1.105 indica a versão do toolkit CUDA instalada.
  • Essa versão pode ser diferente da exibida pelo nvidia-smi, portanto tenha cautela.

Método 3: Verificar o Arquivo version.txt (Verificação Manual)

Se o CUDA estiver instalado em /usr/local/cuda, as informações de versão são registradas no arquivo version.txt.

Comando de Execução

cat /usr/local/cuda/version.txt

Exemplo de Saída

CUDA Version 12.1.105

Pontos Principais

  • Este método é útil caso o comando nvcc -V não esteja disponível.
  • Certifique‑se de que /usr/local/cuda esteja corretamente vinculado à versão desejada do CUDA.

3. Como Verificar a Versão do cuDNN

cuDNN (CUDA Deep Neural Network) é uma biblioteca projetada para deep learning e é usada em combinação com o CUDA.
Além de verificar a versão do CUDA, também é importante confirmar a versão do cuDNN.

Método 1: Verificar o Arquivo cudnn_version.h

A versão do cuDNN está registrada no arquivo de cabeçalho cudnn_version.h.

Comando de Execução

cat /usr/local/cuda/include/cudnn_version.h | grep CUDNN_MAJOR -A 2

Exemplo de Saída

#define CUDNN_MAJOR 8
#define CUDNN_MINOR 9
#define CUDNN_PATCHLEVEL 1

Pontos Principais

  • Esta saída confirma que o cuDNN 8.9.1 está instalado.
  • Usar o comando grep permite recuperar facilmente as informações da versão do cuDNN.
  • Como o cuDNN deve ser compatível com o CUDA, é importante verificar a combinação correta de versões.

Método 2: Verificar com o Comando dpkg (Para Linux baseado em Debian)

Em Ubuntu e outras distribuições Linux baseadas em Debian, você pode checar a versão do cuDNN instalada usando o comando dpkg.

Comando de Execução

dpkg -l | grep libcudnn

Exemplo de Saída

ii  libcudnn8    8.9.1-1+cuda12.1    amd64    NVIDIA cuDNN Library

Pontos Principais

  • A parte libcudnn8 8.9.1-1+cuda12.1 confirma a versão do cuDNN instalada (8.9.1).
  • A parte cuda12.1 indica a versão do CUDA compatível (12.1).

Usando esses métodos, você pode garantir que seu ambiente CUDA esteja configurado corretamente.

4. Como Lidar com Múltiplas Versões CUDA Instaladas

Em um ambiente Ubuntu, várias versões do CUDA podem ser instaladas. No entanto, isso pode gerar confusão sobre qual versão está ativa no momento.
Nesses casos, é necessário mudar para a versão apropriada.

Método 1: Alternar Usando update-alternatives

No Ubuntu, você pode usar update-alternatives para alternar entre versões do CUDA.

Verificar Configurações Atuais

update-alternatives --query cuda

Alternar Versão do CUDA

sudo update-alternatives --config cuda

Exemplo de Saída

There are 3 choices for the alternative cuda (providing /usr/local/cuda).

  Selection    Path                Priority   Status
------------------------------------------------------------
* 0            /usr/local/cuda-11.8  100       auto mode
  1            /usr/local/cuda-10.2  50        manual mode
  2            /usr/local/cuda-11.8  100       manual mode
  3            /usr/local/cuda-12.1  110       manual mode

Press <enter> to keep the current choice[*], or type selection number:

Pontos Principais

  • Executar update-alternatives --config cuda exibirá uma lista das versões do CUDA disponíveis.
  • Você pode selecionar a versão desejada digitando o número correspondente.
  • Existem os modos auto e manual; escolha manual se quiser mudar de versão manualmente.

Método 2: Definir Manualmente um Link Simbólico

Você também pode alternar versões do CUDA modificando o link simbólico.

Verificar Link Simbólico Existente

ls -l /usr/local/cuda

Exemplo de Saída

lrwxrwxrwx 1 root root 20 Feb  1 12:34 /usr/local/cuda -> /usr/local/cuda-11.8

Alterar Versão do CUDA

sudo rm /usr/local/cuda
sudo ln -s /usr/local/cuda-12.1 /usr/local/cuda

Verificar Alteração

ls -l /usr/local/cuda

Pontos Principais

  • /usr/local/cuda serve como o caminho padrão do CUDA. Alterar esse link troca a versão do CUDA.
  • Usando o comando ln -s, você pode mudar a versão do CUDA facilmente, sem precisar modificar configurações globais do sistema.

Com esses métodos, você pode gerenciar eficientemente múltiplas versões do CUDA e garantir que está usando a versão correta para suas necessidades.

5. Perguntas Frequentes (FAQ)

Aqui estão algumas perguntas comuns relacionadas à verificação da versão do CUDA. Se você encontrar algum problema, consulte estas soluções.

Q1: Comando nvcc -V Não Encontrado!

Se o comando nvcc não for encontrado, o CUDA pode não estar instalado corretamente ou seu caminho não está configurado.

Solução 1: Verificar se o CUDA Está Instalado

ls /usr/local/cuda/

Solução 2: Adicionar nvcc ao PATH

export PATH=/usr/local/cuda/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH

Depois de executar esses comandos, tente novamente nvcc -V para verificar se a versão é exibida corretamente.

Q2: Por Que a Versão do CUDA Exibida por nvidia-smi é Diferente?

A versão do CUDA exibida por nvidia-smi representa a versão máxima do CUDA suportada pelo driver NVIDIA, não necessariamente a versão do toolkit CUDA instalada.

Como Verificar:

nvidia-smi

Exemplo de Saída:

CUDA Version: 12.1

Para verificar a versão realmente instalada do CUDA, use nvcc -V ou consulte o arquivo version.txt.

Q3: Como Verificar a Compatibilidade entre CUDA e cuDNN?

A melhor forma de verificar a compatibilidade entre CUDA e cuDNN é consultar a matriz de suporte oficial da NVIDIA.

Documentação Oficial:

NVIDIA cuDNN Support Matrix

Além disso, você pode verificar as versões instaladas usando os seguintes comandos:

Verificar Versão do CUDA

nvcc -V

Verificar Versão do cuDNN

cat /usr/local/cuda/include/cudnn_version.h | grep CUDNN_MAJOR -A 2

Gerenciando seu ambiente corretamente, você pode evitar problemas de compatibilidade entre CUDA e cuDNN.

6. Resumo

Neste artigo, explicamos como verificar a versão do CUDA em um ambiente Ubuntu.
Vamos revisar os pontos principais.

Formas de Verificar a Versão do CUDA

MethodCommandDescription
nvidia-sminvidia-smiShows the CUDA version supported by the NVIDIA driver
nvcc -Vnvcc -VShows the actual installed CUDA toolkit version
version.txtcat /usr/local/cuda/version.txtManually check the CUDA version

Formas de Verificar a Versão do cuDNN

MethodCommandDescription
cudnn_version.hcat /usr/local/cuda/include/cudnn_version.h | grep CUDNN_MAJOR -A 2Check the version from the header file
dpkg Commanddpkg -l | grep libcudnnCheck the installed cuDNN version

Como Alternar Versões do CUDA

MethodCommandDescription
update-alternativessudo update-alternatives --config cudaSwitch between multiple CUDA versions
Symbolic Linksudo ln -s /usr/local/cuda-XX.X /usr/local/cudaManually change the CUDA version

Principais Pontos Chave

  • É importante identificar corretamente a versão do CUDA
  • Garanta a compatibilidade entre CUDA e cuDNN
  • Se estiver usando múltiplas versões do CUDA, entenda como alternar entre elas

Gerenciando seu ambiente adequadamente, você pode maximizar os benefícios do CUDA.
Esperamos que este artigo ajude você a verificar a versão do CUDA em seu ambiente Ubuntu.

Artigos Relacionados

侍エンジニア塾