CUDA (+CUDA Toolkit), Nvidia Driver, cuDNN 우분투 20.04 설치
Reference: 출처 1, 출처 2, CUDA 공식 웹사이트 (버전 체크), 공식으로 나온 설치 방법 파이토치와 버전을 맞추기 위해 cuda11.1, cuDNN 8.1.1을 설치하겠습니다. 최근에는 cuda11.3까지 호환이 되어 cuda11.3, cuDNN 8.2.0이나 8.2.1을 설치해도 되는 모양입니다. 설치방법은 거의 동일하고 cuda 버전만 11.3으로 적어주면 됩니다. 관련 링크도 같이 첨부하였습니다. [기존 파일 지우기] cuda를 설치한 적이 있다면 apt에 repo가 등록되어있을 건데 지워줍니다. 등록된 key값 있는지 확인 sudo apt-key list nvidia/cuda 패키지가 존재한다면 sudo rm /etc/apt/sources.list.d/cuda* nvidia-cud..
PyTorch
2021. 3. 2. 20:43