ubuntu安裝cuda和cudnn

ubuntu18.04+cuda11.1+cudnn8

卸載

卸載cuda（根據自己的版本來）

sudo /usr/local/cuda-10.2/bin/cuda-uninstaller

或者

sudo apt-get --purge remove "*cublas*" "*cufft*" "*curand*" "*cusolver*" "*cusparse*" "*npp*" "*nvjpeg*" "cuda*" "nsight*"
sudo apt-get remove cuda
sudo rm -r cuda-11.1

卸載cudnn（根據自己的版本來）

deb安裝

sudo apt-get remove libcudnn8-samples
sudo apt-get remove libcudnn8-dev
sudo apt-get remove libcudnn8

通過壓縮包安裝的cudnn隨着cuda卸載

deb版本

安裝前卸載之前安裝的cuda和cudnn

安裝cuda

下載地址

官網命令

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.1/local_installers/cuda-repo-ubuntu1804-11-1-local_11.1.1-455.32.00-1_amd64.deb
sudo dpkg -i cuda-repo-ubuntu1804-11-1-local_11.1.1-455.32.00-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

添加環境變量（/etc/profile或者~/.bashrc）

vim ~/.bashrc

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

刷新

source ~/.bashrc

驗證

nvcc -V

安裝cudnn

下載地址

sudo dpkg -i libcudnn8_8.0.2.39-1+cuda10.2_amd64.deb 
sudo dpkg -i libcudnn8-dev_8.0.2.39-1+cuda10.2_amd64.deb
sudo dpkg -i libcudnn8-doc_8.0.2.39-1+cuda10.2_amd64.deb

查看版本

cat /usr/include/cudnn_version.h

驗證

cp -r /usr/src/cudnn_samples_v8/ $HOME
cd ~/cudnn_samples_v8/mnistCUDNN/
make clean && make
./mnistCUDNN

結果：Test passed!

run版本

安裝cuda

sudo chmod a+x cuda_11.0.2_450_51.05_linux.run
sudo ./cuda_11.0.2_450_51.05_linux.run

選項中去掉安裝驅動的選項

安裝顯卡驅動會默認安裝對應版本的cuda，此時會提示已經存在cuda。 繼續安裝

是否接收協議。 接收

出現一系列安裝提示。 不安裝顯卡驅動

是否生成/更新動態鏈接/usr/local/cuda。 更新

如果出現動態鏈接庫鏈接不到的情況，建立軟連接

sudo ln -sf /usr/local/cuda-9.0/targets/x86_64-linux/lib/libcudnn.so.7.4.2 /usr/local/cuda-9.0/targets/x86_64-linux/lib/libcudnn.so.7

添加環境變量（/etc/profile或者~/.bashrc）

vim ~/.bashrc

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

刷新

source ~/.bashrc

驗證

cd /usr/local/cuda/samples/1_Utilities/deviceQuery 
sudo make
./deviceQuery

結果Result = PASS

安裝cudnn

使用編譯好的壓縮包形式

解壓對應版本的壓縮包

sudo cp cuda/include/cudnn*.h /usr/local/cuda/include
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64
sudo chmod a+r /usr/local/cuda/include/cudnn*.h /usr/local/cuda/lib64/libcudnn*

驗證&查看版本

cat /usr/local/cuda/include/cudnn_version.h | grep CUDNN_MAJOR -A 2
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2