ubuntu18.04安裝cuda11.1

安裝

wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/cuda-ubuntu1804.pin
sudo mv cuda-ubuntu1804.pin /etc/apt/preferences.d/cuda-repository-pin-600
wget https://developer.download.nvidia.com/compute/cuda/11.1.0/local_installers/cuda-repo-ubuntu1804-11-1-local_11.1.0-455.23.05-1_amd64.deb
sudo dpkg -i cuda-repo-ubuntu1804-11-1-local_11.1.0-455.23.05-1_amd64.deb
sudo apt-key add /var/cuda-repo-ubuntu1804-11-1-local/7fa2af80.pub
sudo apt-get update
sudo apt-get -y install cuda-11.1

在bashrc添加

export PATH="/usr/local/cuda-11.1/bin:$PATH"
export LD_LIBRARY_PATH="/usr/local/cuda-11.1/lib64:$LD_LIBRARY_PATH"
export PATH="/usr/local/cuda/bin:$PATH"

source .bashrc

python -m pip install pycuda

卸載cuda

cd /usr/local/cuda/bin
sudo ./cuda-uninstaller

報錯
/etc/apt/preferences.d/cuda-repository-pin-600 中發現有無效的記錄，無 Package 字段頭 刪除掉

 cd /etc/apt/preferences.d
 rm cuda-repository-pin-600