ubuntu 16.04安装CUDA9.0 CUNN7.05

1、下载CUDA9.0

打开下载地址:https://developer.nvidia.com/cuda-90-download-archive

2、下载后进入对应文件夹,输入sudo sh cuda_9.0.176_384.81_linux.run

然后一直Enter到100%,并输入accept

接着出现 Install NVIDIA Accelerated Graphics Driver for Linux-x86_64 384.81?

一定要选择No,否则之前的驱动安装就白安了

3、添加环境变量

sudo gedit /etc/profile

打开“profile”文件,在末尾处添加(注意不要有空格,不然会报错):

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

重启电脑

4、输入nvcc -V,测试是否安装成功

5、安装CUDNN v7.05

下载地址:https://developer.nvidia.com/rdp/cudnn-archive#a-collapse705-9

6、下载后解压文件,解压之后,得到一个 cudn 文件夹,该文件夹下include 和 lib64 两个文件夹,命令行进入 cuda/include 路径下,然后进行以下操作:

cd cuda/include
sudo cp cudnn.h /usr/local/cuda/include #复制头文件

再进入lib64目录下的动态文件进行复制和链接:

cd ..
cd lib64
sudo cp lib* /usr/local/cuda/lib64/ #复制动态链接库
cd /usr/local/cuda/lib64/
sudo rm -rf libcudnn.so libcudnn.so.7 #删除原有动态文件
sudo ln -s libcudnn.so.7.0.5 libcudnn.so.7 #生成软衔接
sudo ln -s libcudnn.so.7 libcudnn.so #生成软链接
sudo ldconfig

验证CUDNN是否安装成功:

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

最后推荐一个ubuntu下的截图软件,方便写博客使用:Shutter

参考地址:https://jingyan.baidu.com/article/4b52d702db5482fc5c774bf0.html

猜你喜欢

转载自blog.csdn.net/oMoDao1/article/details/90299228
今日推荐