环境:Ubuntu 20.04 x86_64(其他版本也可以在官网参考安装)
Download CUDA for Linux Ubuntu 20.04 x86_64
wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/cuda-ubuntu2004.pin
sudo mv cuda-ubuntu2004.pin /etc/apt/preferences.d/cuda-repository-pin-600wget https://developer.download.nvidia.com/compute/cuda/11.1.0/local_installers/cuda-repo-ubuntu2004-11-1-local_11.1.0-455.23.05-1_amd64.deb
sudo dpkg -i cuda-repo-ubuntu2004-11-1-local_11.1.0-455.23.05-1_amd64.deb
sudo apt-key add /var/cuda-repo-ubuntu2004-11-1-local/7fa2af80.pub
sudo apt-get update
sudo apt-get -y install cuda
看一下安装成功没有:
nvidia-smi
这样就说明可以了
添加环境变量
sudo vim ~/.bashrc
下面的/usr/local/cuda-11.1/bin替换成你自己的cuda11的bin目录
同样/usr/local/cuda-11.1/lib64也是替换成自己的cuda11的lib64目录
export PATH=/usr/local/cuda-11.1/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-11.1/lib64$LD_LIBRARY_PATH
如果nvcc -V不是这样的那么可能是环境变量没设置好