我们的征途是星辰大海

ubuntu安装cuda9.0和cudnn

关于cuda

1. 首先确保nvidia驱动已安装(在设置-软件与更新-附加驱动里更改重启即可)
2. 官网直接下载所需cuda版本(如果无法直接找到旧版,可直接谷歌cuda版本)选择如下

《ubuntu安装cuda9.0和cudnn》

3. 确保对应gcc,g++版本适合cuda版本

CUDA 8开始,Ubuntu 16.06支持gcc 5.3

CUDA 9开始,Ubuntu 16.04,Ubuntu 17.04支持gcc 6

我的gcc版本是5.4,所以装cuda9.0完全没有问题,如果需要安装cuda8,则需降级gccg++版本。

查看当前gcc g++版本
gcc --version
g++ --version

4. 直接在terminal中运行sh cuda_name.sh

在安装驱动这里选择no其余accept yes

5. 加入环境变量

#在.bashrc文件中末尾添加

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

更新环境变量

source ~/.bashrc

6. 测试

nvcc --version
《ubuntu安装cuda9.0和cudnn》

关于cudnn

1. 选择文件下载

我选择的是library for linux

2. 解压复制

解压后得到一个“cuda”文件夹,将其中文件复制到cuda安装的对应文件夹下,我的路径是/usr/local/cuda
注意文件夹权限

3. 测试

cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
《ubuntu安装cuda9.0和cudnn》

成功!

原文链接: https://www.astarsea.com/ubuntu安装cuda9-0和cudnn/

点赞

发表评论

电子邮件地址不会被公开。 必填项已用*标注