您当前的位置:首页 > 计算机 > 系统应用 > Linux

Ubuntu16.04 安装CUDA和CUDNN、CUDA多版本替换

时间:07-01来源:作者:点击数:
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

以上步骤1和步骤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
方便获取更多学习、工作、生活信息请关注本站微信公众号城东书院 微信服务号城东书院 微信订阅号
推荐内容
相关内容
栏目更新
栏目热门