envs/Anaconda

[Anaconda] CUDA 버전 체크

제이스핀 2022. 1. 12. 01:32
반응형

1. nvidia-smi

 

2. nvcc --version

CUDA는 2개의 API를 가지고 있는데, 하나는 runtime API이고 다른 하나는 driver API이다.

각 API가 각자의 version을 가지고 있다.

 

1. nvidia-smi 의 CUDA는 runtime API를 나타내며, GPU 드라이버에 의해 설치된다.

2. nvcc의 CUDA는 CUDA toolkit에 의해 설치된다.

 

설치한 CUDA 버전을 확인하는건 2. nvcc --version 명령어가 옳다.

 

 

 

근데 여기서, 분명히 conda에 다른 버전의 cudatoolkit을 설치했는데,

nvcc --version을 할 시 이전 cuda 버전이 계속 나온다?

 

'nvcc --version is not working in anaconda prompt if you have the cuda toolkit installed with conda.'

 

 

nvcc --version은 conda 내에 설치된 cuda가 아닌, base의 cuda 버전을 출력한다.

 

 

3. conda list cudatoolkit

 

위 명령어를 통해 해당 conda에 설치된 cudatoolkit 버전을 확인할 수 있다.

 

 

conda info 로도 확인 가능하다.

 

 

반응형