我们的征途是星辰大海

0%

ubuntu安装cuda9.0和cudnn

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++版本。

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
查看当前gcc g++版本
gcc --version
g++ --version
````


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


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

**5. 加入环境变量**
```` bash
#在.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

更新环境变量

1
source ~/.bashrc

6. 测试

1
nvcc --version

关于cudnn

1. 选择文件下载

我选择的是library for linux

2. 解压复制

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

3. 测试

1
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
成功!