cuda11.1 + cuDNN v8.0.4 for CUDA 11.1 配置流程

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

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博客

最后编辑于 ：
©著作权归作者所有,转载或内容合作请联系作者
平台声明：文章内容（如有图片或视频亦包括在内）由作者上传并发布，文章内容仅代表作者本人观点，简书系信息发布平台，仅提供信息存储服务。