查看自己的驱动
$ ubuntu-drivers devices
依赖库的安装
$ sudo apt-get install freeglut3-dev build-essential libx11-dev libxmu-dev
$ sudo apt-get install libxi-dev libgl1-mesa-glx libglu1-mesa libglu1-mesa-dev
安装cuda9.0
进入这个地址下载:
然后在网页中往下拉,下载下图中的那个:
image.png
下载后,进入你放这这个文件的文件夹,在此文件夹下操作。
$ sudo dpkg -i cuda-repo-ubuntu1704-9-0-local_9.0.176-1_amd64.deb
$ sudo apt-key add /var/cuda-repo-9-0-local/7fa2af80.pub
如果以上成功运行的话,会返回OK。
$ sudo apt-get update
$ sudo apt-get install cuda
到这里就应该是安装完了,然后检查一下是否安装成功。
$ cd /usr/local
$ ls
看到下面的文件夹有cuda和cuda-9.0就是我们已经安装成功了。
环境配置
打开终端输入:
$ sudo gedit ~/.bashrc
在打开的配置文件中,将以下内容写入末尾处.
export CUDA_HOME=/usr/local/cuda-9.0
export LD_LIBRARY_PATH=/usr/local/cuda-9.0/lib64:/usr/local/cuda-9.0/extras/CUPTI/lib64:$LD_LIBRARY_PATHs
export PATH=/usr/local/cuda-9.0/bin:$PATH
export LD_LIBRARY_PATH="/usr/local/cuda-9.0/lib64:/usr/local/cuda-9.0/extras/CUPTI/lib64:/usr/local/cuda-9.0/lib64:/usr/local/cuda-9.0/lib64"
使环境变量生效.
$ source ~/.bashrc
测试
$ cd /usr/local/cuda-9.0/samples/1_Utilities/deviceQuery
$ sudo make
$ ./deviceQuery
如果结果有GPU的信息,说明安装成功。
安装cudnn7.0
为适应Cudn9.0,这里下载的是CuDnnv7.0.5, for CUDA 9.0(cuDNN v7.0.5 Library for Linux)。
下载地址:https://developer.nvidia.com/rdp/cudnn-archive
解压并复制相应文件
$ tar -xvzf cudnn-9.0-linux-x64-v7.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*
安装后进行测试
$ nvcc --version
失败的话重启再试试。
网友评论