[Python] VSCODE에서 원하는 버전의 Cuda-toolkit 설치

vscode에서 가상환경 만든 후, cuda-toolkit 원하는 버전으로 설치하려고 할 때 에러가 발생. 

이렇게 가상환경에 원하는 버전의 cuda-toolkit을 만들기를 원하는 이유는 torch geometric을 사용하려고 할 때 cuda 11.8까지밖에 지원이 안되기 때문! 

새로운 버전이 출시되면서 업데이트가 되긴 하지만, 속도가 좀 느리다 ㅠㅠ 그래서 최근에 컴퓨터를 구입한 사람들의 경우는 낮은 버전의 cudatoolkit을 설치해야함. 

 

 

그런데 사이트에서 제공하는 다운로드 코드를 그대로 사용할 경우, 이런 에러가 발생한다.

The following packages have unmet dependencies:
 cuda-11-8 : Depends: cuda-runtime-11-8 (>= 11.8.0) but it is not going to be installed
             Depends: cuda-demo-suite-11-8 (>= 11.8.86) but it is not going to be installed
E: Unable to correct problems, you have held broken packages.

 

구글링해서 apt clean, apt update 요런 조치들을 하라고 해서 시도해도 여전히 같은 에러가 발생함. 

그래서 코드를 이렇게 바꿔서 하면 에러가 없어진다

일단 nvidia 공식 사이트에서 제공하는 코드는 이렇다:

wget https://developer.download.nvidia.com/compute/cuda/11.8.0/local_installers/cuda-repo-debian11-11-8-local_11.8.0-520.61.05-1_amd64.deb
sudo dpkg -i cuda-repo-debian11-11-8-local_11.8.0-520.61.05-1_amd64.deb
sudo cp /var/cuda-repo-debian11-11-8-local/cuda-*-keyring.gpg /usr/share/keyrings/
sudo add-apt-repository contrib
sudo apt-get update
sudo apt-get -y install cuda

 

내가 사용한 코드는 이렇다:

wget https://developer.download.nvidia.com/compute/cuda/11.8.0/local_installers/cuda-repo-debian11-11-8-local_11.8.0-520.61.05-1_amd64.deb
sudo dpkg -i cuda-repo-debian11-11-8-local_11.8.0-520.61.05-1_amd64.deb
sudo cp /var/cuda-repo-debian11-11-8-local/cuda-*-keyring.gpg /usr/share/keyrings/
sudo apt-key adv --fetch-keys http://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/7fa2af80.pub
sudo add-apt-repository "deb https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/ /"
sudo apt update
sudo apt install cuda-11-8

 

에러없이 완료되는 화면을 볼 수 있음..