Ubuntu16.04在cuda10.2的基础上安装cuda11.0并存
前言
在ubuntu16.04系统中,安装了cuda10.2版本,但是yolov5.5对应的libtorch版本为1.7.0,需要搭配cuda11.0环境,因此需要在cuda10.2的基础上安装cuda11.0并可进行版本切换.
一、安装cuda11.0
因为之前已安装显卡驱动,最高可支持cuda11.2版本,因此跳过安装NVIDIA显卡驱动部分.
需要的可参考链接:Ubuntu18.04下NVIDIA驱动+CUDA11.0安装
cuda官方安装包链接(包含所有版本):
https://developer.nvidia.com/cuda-toolkit-archive
选择CUDA Toolkit 11.0 Update1 (Aug 2020), Versioned Online Documentation版本进行下载.
按照自己的配置进行选择:
通过下面的命令下载并运行安装:
wget https://developer.download.nvidia.com/compute/cuda/11.0.3/local_installers/cuda_11.0.3_450.51.06_linux.run
sudo sh cuda_11.0.3_450.51.06_linux.run
在最后一行输入:accept
因为已经安装过驱动,所以把驱动那一栏取消勾选(按"enter")
按向下的方向键到最后一行"install",点击enter键.
其他的都选择同意即可,最后完成安装,但是会提示更新环境变量:
打开.bashrc文件:
gedit ~/.bashrc
添加下面两行,或者把之前存在的10.2更改为11.0:
export PATH=/usr/local/cuda-11.0/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-11.0/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
验证安装是否成功:
nvcc -V
显示cuda版本为11.0:
二、安装cuDNN v8.0.4
cudnn官方安装包链接(包含所有版本):
https://developer.nvidia.com/rdp/cudnn-archive
提示:需要进行注册登录才可以下载,可使用谷歌邮箱注册.
这里选择cuDNN Library for Linux (x86_64)版本进行下载.
代码如下（示例）：
参考链接:
Ubuntu16.04离线安装cuda11.1和cudnn8.0.4
把下载的文件解压:
tar -xvf cudnn-11.1-linux-x64-v8.0.4.30.tgz
进入解压的文件夹
sudo cp ./cuda/include* /usr/local/cuda/include/ //看了一篇博客说一定要这样复制,很多是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
sudo chmod a+r /usr/local/cuda/lib64/libcudnn*
后记
更改./bashrc文件中的两行代码对应的版本即可更改cuda版本。
有人说需要更改软连接,我没有做这一步,可以参考链接:
https://blog.csdn.net/sazass/article/details/119832566