1.安装cuda11.1
- 在nvidia官网下载与445.38驱动版本对应的cuda版本11.1.1,地址为https://developer.nvidia.com/cuda-toolkit-archive。建议采用runfile(local)方式下载安装。安装cuda时,不要选中安装driver,其余都选中安装.
wget https://developer.download.nvidia.com/compute/cuda/11.1.1/local_installers/cuda_11.1.1_455.32.00_linux.runsudo
sh cuda_11.1.1_455.32.00_linux.run
- 打开主目录下的 .bashrc文件添加如下路径,.bashrc文件在/home下,如果没有找到,则按Ctrl+H键显示隐藏文件。将以下命令添加至文件末尾,保存退出。
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
注:上述命令中的cuda也可以改为cuda11.1,因为在系统目录中有cuda和cuda11.1两个文件夹,其中cuda文件夹是cuda11.1的链接,二者内容相同。
- 检查:nvcc -V,查看cuda是否正确安装
2.安装cudnn8.0.4
- 在NVIDIA官网下载与cuda11.1版本对应的cudnn8.0.4,下载地址为:https://developer.nvidia.com/rdp/cudnn-archive。建议下载cuDNN Library for Linux (x86_64)安装包。
- 安装cudnn v8.0.4
sudo cp cudnn-11.1-linux-x64-v8.0.4.30.solitairetheme8 cudnn-11.1-linux-x64-v8.0.4.30.tgz
tar -xzvf cudnn-11.1-linux-x64-v8.0.4.30.tgz
sudo cp cuda/include/* /usr/local/cuda/include/
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64/
sudo chmod a+r /usr/local/cuda/include/cudnn.h
sudo chmod a+r /usr/local/cuda/lib64/libcudnn*
注:此处必须使用sudo cp cuda/include/* /usr/local/cuda/include/,而不是sudo cp cuda/include/cudnn.h /usr/local/cuda/include/
https://developer.nvidia.com/tensorrt-download
安装TensorRT
https://blog.csdn.net/Williamcsj/article/details/123514435
卸载cuda
(108条消息) Ubuntu18.04安装CUDA10.2 + cuDNN v7.6.5_坚持健身的程序员的博客-CSDN博客