ubuntu安装cuda9.0和cudnn

关于cuda

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

关于cudnn

1.选择文件下载
我选择的是library for linux
2.解压复制
解压后得到一个“cuda”文件夹,将其中文件复制到cuda安装的对应文件夹下,我的路径是/usr/local/cuda
注意文件夹权限
3.测试
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
成功!
Tags:

Add a Comment

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