环境搭建02-Ubuntu16.04 安装CUDA和CUDNN、CUDA多版本替换

最新推荐文章于 2025-04-16 09:06:24 发布

JLGao的博客

最新推荐文章于 2025-04-16 09:06:24 发布

阅读量870

点赞数 3

分类专栏：深度学习环境搭建(Ubuntu16.04) 文章标签： linux 深度学习

本文链接：https://blog.youkuaiyun.com/qq_40541102/article/details/129280305

版权

深度学习环境搭建(Ubuntu16.04) 专栏收录该内容

13 篇文章

订阅专栏

1、CUDA安装

（1）下载需要的CUDA版本
https://developer.nvidia.com/cuda-toolkit-archive

（2）安装

sudo sh cuda_8.0.61_375.26_linux.run

（3）添加环境

gedit ~/.bashrc

在文件末尾添加：

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

（4）使文件有效

source ~/.bashrc

（5）查看安装是否成功

cat /usr/local/cuda-8.0/version.txt

或者

nvcc -V   ## nvcc -version

2、CUDNN安装

（1）下载CUDNN
https://developer.nvidia.com/cudnn
需要注册用户，然后登录下载。注意cudnn的版本要和cuda的版本对应。

（2）安装

sudo cp cuda/include/cudnn.h /usr/local/cuda-8.0/include/
sudo cp cuda/lib64/libcudnn* /usr/local/cuda-8.0/lib64/
sudo chmod a+r /usr/local/cuda-8.0/include/cudnn.h
sudo chmod a+r /usr/local/cuda-8.0/lib64/libcudnn*

（3）查看是否安装成功

cat /usr/local/cuda-8.0/include/cudnn.h | grep CUDNN_MAJOR -A 2

以上步骤１和步骤２就是但版本CUDA的安装

3、CUDA多版本替换

例如：将CUDA-8.0更换为CUDA-10.0。这里，CUDA-8.0和CUDA-10.0以正确安装完成。可以通过创建cuda软链接指向不同版本的cuda版本。
首先，修改CUDA的路径，

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

这里cuda软链接指向cuda-8.0版本。
然后，删除指向cuda-8.0的软链接cuda，重新创建一个指向cuda-10.0版本的软链接。

rm -rf /usr/local/cuda
sudo ln -s /usr/local/cuda-10.0 /usr/local/cuda
nvcc --version