nvidia-smi 와nvcc -V의 차이

cmd 창에서 보이는 nvidia-smi는 이 노트북(local)에서 최대 이 버전까지 지원해준다는 의미이다.(업데이트하면 또 달라짐.)

nvcc-V는 지금 이 노트북(local)에 깔린 cudnn이 몇 버전인지에 대한 명령어이다.

가상환경에 pytorch깔 때는 어떤 버전이든 상관없다. 단, nvidia-smi로 했을 때 나오는 버전 아래로만 깔면 된다. 이때 설치한 pytorch버전이 cudnn과 호환이 안 될 수도 있다. (안되면 찾아봐야한다!)

 

🧐 중요

가상환경 안에서 pytorch 특정 버전 설치해도 nvidia-smi, nvcc -V해서 나온 버전은 local의 버전이라서 가상환경과는 아무 관련 없다.

즉, 가상환경에서 원하는 버전 설치하면 된다.

가상환경에 깔린 cuda version 확인하려면

python
>>> import torch
>>> torch.__version__
# torch 버전
>>> torch.version.cuda
# cuda 버전 확인