Ubuntu16.04下安装cuda和cudnn

一般而言,现在最便捷的方式是在anaconda 下用conda 安装 ,但是在用rcnn或其他时,需要手动安装cuda和cudnn,这里介绍手动安装的办法,主要参考csdn链接:https://blog.csdn.net/wanzhen4330/article/details/81699769

Nvidia安装指南

https://docs.nvidia.com/deeplearning/sdk/cudnn-install/index.html#axzz4qYJp45J2

1.cuda

已安装驱动的情况:

英伟达官网下载对于的cuda

https://developer.nvidia.com/cuda-toolkit-archive

cd 到下载目录下

运行命令:

sudo sh cuda_9.0.176_384.81_linux.run

指挥出出现more0%的字样,一直按住回车,知道100%

接下来依次选择

accept,

n(不安装驱动)

y

y

y

安装完成后设置换环境变量

(这里进行了改动,csdn上的路径并不准确)

在home/用户名  目录下打开.bashrc文件,注意,用winscp无法修改,所以建议在图形界面打开。

用ctrl+H显示隐藏文件,在.bashrc的末尾加上以下命令即可添加路径:

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

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

export CUDA_HOME=/usr/local/cuda-9.0

终端运行:source ~/.bashrc

检查:nvcc --version,如果显示下面的文字就说明安装成功了。


2.cudnn

先用conda list查看已安装的cuda和cudnn版本,避免冲突,

cudnn的安装文件:https://developer.nvidia.com/rdp/cudnn-archive

之后用谷歌浏览器,别用edge,否则不是tgz格式的压缩包

我的cuda对应的时cudnn7.3.1版本,以此为例


下载标线的四个文件,后三个deb文件只有16.04或者18.04能用(手动忽略14。04),其他版本不需要安装

先解压第一个文件

tar -xvf cudnn-9.0-linux-x64-v.7.3.1.20.tgz

复制路径:

$ sudo cp cuda/include/cudnn.h /usr/local/cuda/include

$ sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64

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

之后以此安装三个deb

sudo dpkg -i libcudnn7_7.3.1.20-1+cuda9.0_amd64.deb

sudo dpkg -i libcudnn7-dev_7.3.1.20-1+cuda9.0_amd64.deb

sudo dpkg -i libcudnn7-doc_7.3.1.20-1+cuda9.0_amd64.deb

输入命令

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

如果出现以下界面说明安装成功



在linux下,子用户想用cuda和cudnn只需找到自己目录下的.bashrc文件,加入路径即可.

也就是在.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

你可能感兴趣的:(Ubuntu16.04下安装cuda和cudnn)