在ubuntu中安装CUDA和cuDNN

安装前确保系统中支持CUDA的Nvidia显卡和gcc编译器

安装CUDA

下载CUDA10.1.105_418，执行如下命令：

sudo dpkg -i cuda-repo-ubuntu1810-10-1-local-10.1.105-418.39_1.0-1_amd64.deb
/*执行这条命令打印出的内容
Selecting previously unselected package cuda-repo-ubuntu1810-10-1-local-10.1.105-418.39.
(Reading database ... 186150 files and directories currently installed.)
Preparing to unpack cuda-repo-ubuntu1810-10-1-local-10.1.105-418.39_1.0-1_amd64.deb ...
Unpacking cuda-repo-ubuntu1810-10-1-local-10.1.105-418.39 (1.0-1) ...
Setting up cuda-repo-ubuntu1810-10-1-local-10.1.105-418.39 (1.0-1) ...

The public CUDA GPG key does not appear to be installed.
To install the key, run this command:
sudo apt-key add /var/cuda-repo-10-1-local-10.1.105-418.39/7fa2af80.pub
*/
sudo apt-key add /var/cuda-repo-10-1-local-10.1.105-418.39/7fa2af80.pub
sudo apt-get update
sudo apt-get install cuda

，重启

检查CUDA的安装情况

nvidia-smi

将nvcc添加到环境变量中

vim ~/.bashrc
# 添加一行：export PATH="/usr/local/cuda-10.1/bin:$PATH"
source ~/.bashrc

，执行nvcc -V

安装cuDNN

下载cudnn-10.1-linux-x64-v7.6.5.32.tgz(cudnn for linux)

tar -xzvf cudnn-10.1-linux-x64-v7.6.5.32.tgz
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* # 所有用户组赋上读权限
vim ~/.bashrc
# 添加一行：export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH
source ~/.bashrc

摘编自ubuntu安装cuda、cudnn和nvidia-docker

本文所需软件包可于下载页中下载

本文创建于2021年7月19日14点12分，修改于2021年7月19日14点12分