cuda 버전

nvidia-smi: 드라이버가 지원하는 최대 CUDA 버전

nvcc -V: 실제로 설치된 CUDA 버전

 

torch.version.cuda 이거말고 이거 nvidia-smi고 이버전 믿지말고 nvcc -V로 나온 버전이 진짜 cuda 버전(설치된 cuda 버전)이에요.

 

 

 

 

cuda 버전 맞춰주기 위해

ubuntu cuda 다시 설치

ubuntu 20.04, nvidia driver 495, cuda 11.3, cudnn 8.2.1, pytorch 버전을 설치하는 방법입니다.
우분투에서 파이썬 가상환경도 만들어봅니다.

 

기존 nvidia, cuda 완전 삭제

• 혹시 우분투에 nvidia, cuda를 설치한 적이 있다면 해당 명령어를 터미널에 실행해보길 바란다. 1

    $ sudo apt-get autoremove
  

    $ sudo apt-get autoclean
  

    $ sudo rm -rf /usr/local/cuda*
  

• $ sudo apt-get purge nvidia*

설치가능한 드라이버 확인

• 그래픽 카드 및 설치 가능한 드라이버를 다음 명령어로 확인해본다.
• $ ubuntu-drivers devices

 

• 이제 설치하고픈 버전을 골라 설치하거나, 자동으로 드라이버를 설치한다. 2
• 본인은 nvidia-driver-495로 설치해보겠다.

    # 원하는 버전 설치
    $ sudo apt install nvidia-driver-495
  

• # 자동으로 드라이버 설치 $ sudo ubuntu-drivers autoinstall

 

• nvidia 드라이버 설치가 끝나면 NVIDIA kernel module의 load를 도와주는 modprobe 패키지를 설치한다. 3
• $ sudo apt-get install dkms nvidia-modprobe
• 그리고 다음 명령어로 마무리해준다.
• $ sudo apt update $ sudo apt upgrade # 재부팅 $ sudo reboot

 

• 재부팅 후 설치가 잘 되었는지 확인하기 위해 다음 명령어를 실행해본다.
• 아래처럼 잘 나오면 설치가 잘 된것이다.
• $ nvidia-smi

 

cuda 11.3 설치

• cuda 11.3을 설치하기 위해 해당 사이트에 들어간다. (11.3 버전으로 설치하면 nvcc -V 명령어 안먹음 11.3.1로 진행)
• NVIDIA DEVELOPER - CUDA Toolkit Archive
• 그 다음, linux -> x86_64 -> Ubuntu -> 20.04 -> runfile(local)로 가 해당 명령어를 실행한다.

 

 

 

근데 nvidia driver 설치 후

이렇게 나옴

구글링하면 해결법 나오는데 그냥 다시 드라이버 지우고 다시 설치하니 nvidia-smi 명령어 잘 나옴.

 

 

 

참고

Different CUDA versions shown by nvcc and NVIDIA-smi