Ubuntu20.04安装cuda11.4.2 cudnn 8.2.4

cuda安装

• 先确认已安装好显卡驱动， Driver Version: 470.82.01 下载地址略

sudo sh cuda_11.4.2_470.57.02_linux.run

• 是否接受EULA？ 输入accept
• 不用选择除了cuda以外的其他选项，特别是驱动不用安装
• 完成后添加环境变量

vim ~/.bashrc

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

source ~/.bashrc

• 没有安装simple就不用测试了

cudnn安装

• 下载地址略
• 解压后复制到指令目录

tar zxvf cudnn-11.4-linux-x64-v8.2.4.15.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*

• 测试，有打印8.2.4安装成功

cat cuda/include/cudnn_version.h |grep ^#

cuda和cudnn卸载

• cuda卸载

cd /usr/local/cuda/bin/
sudo ./cuda-uninstaller
sudo rm -rf /usr/local/cuda-11.0
sudo rm -rf /usr/local/cuda

• cudnn卸载

sudo rm -rf /usr/local/cuda/include/cudnn.h
sudo rm -rf /usr/local/cuda/lib64/libcudnn*

Reference1
Reference2