在安装之前需要先确定已经安装好与显卡适配的驱动,如果没有的话:
- 下载驱动:https://www.nvidia.cn/Download/index.aspx?lang=cn
- 安装:https://blog.csdn.net/frothmoon/article/details/88875020
如果安装好了就可以下面的安装了。
安装CUDA
下载
有一些注意事项可以在安装前看一看https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html
其中有一点需要注意的,安装的CUDA版本需要和驱动版本对应。
此外CUDA可以安装多个版本,根据需要可以进行安装,在使用时注意切换版本。
此处下载与驱动版本匹配的CUDA:https://developer.nvidia.com/cuda-toolkit-archive
在页面中根据自己的实际情况选择不同的系统配置,下载即可。
我安装的是CUDA10.0。
安装
下载后是一个run文件,在下载目录运行sudo sh cuda_xxxxx_linux.run
,此处xxxxx替换为你的型号。
- 一直按空格键到后面,提示接受条款,输入accept接受。
- 然后询问是否安装驱动,因为前面已经安装过了,输入n不安装。
- 接着输入y同意安装cuda工具。
- 输入y安装指向/usr/local/cuda的符号链接。
- 输入y安装samples,可以测试是否安装成功。
- 回车确认samples安装的默认路径,/home/用户名,安装路径测试完可以删除。
添加环境变量
命令sudo gedit /etc/profile
打开一个文件,在最后添加:
export PATH=/usr/local/cuda-10.0/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-10.0/lib64:/usr/local/cuda-10.0/extras/CUPTI/lib64:$LD_LIBRARY_PATH
没有gedit可以用sudo apt-get install gedit
安装。
输入的两行中cuda版本要根据你下载的替换,如cuda-9.0等
保存文件后重启系统。
测试
重启后可以测试CUDA Toolkit安装是否正确:nvcc --version
,输出信息说明安装正确。
接下来编译samples。
默认路径为~/NVIDIA_CUDA-10.0Samples
,命令行cd
进入后make
,这里需要等一段时间。
之后会生成可执行文件在~/NVIDIA_CUDA-10.0_Samples/bin/x84_64/linux/release
,进入目录后输入./deviceQuery
。
会有一堆输出。
这就安装好啦。
安装cuDNN
cuDNN的安装非常简单,但是下载需要注册账号登录。
官网下载:https://developer.nvidia.com/rdp/cudnn-archive,下载时注意选择与CUDA适配的版本,点开后直接下载for Linux的就可以。
这里用windows下载的文件格式和ubuntu下载的文件格式不一样,但windows下载的solitairetheme8格式文件可以重命名为和ubuntu一样的tar.gz文件。
我安装的是7.6.5版本。
安装时用tar -xvzf cudnn-10.0-linux-x64-vxxxx.tar.gz
进行解压,其中xxxx替换为你的版本。
然后进行一些文件的复制和权限授予。
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*
安装完成。
参考教程:http://ttshun.com/2018/09/08/Caffe%E7%9A%84GPU%E6%A8%A1%E5%BC%8F%E5%AE%89%E8%A3%85/