一:CUDA安装
1.nvidia官网下载cuda
直接搜索cuda,https://developer.nvidia.com/cuda-downloads https://developer.nvidia.com/cuda-toolkit-archive
选择历史版本,下载自己想要的版本。
本人下载的这个: CUDA Toolkit 10.0 (Sept 2018), Online Documentation,点击进入
本人选择的ubuntu1604 x86_64 runfile[local]这一项安装方式,其他选项也可是,看个人喜好用哪种方式安装。
这个cuda有两个文件,有的有多个,不一定,都下载下来,一个是主程序,剩下的都是补丁。
chmod +x cuda_10.0.130.1_linux.run
chmod +x cuda_10.0.130_410.48_linux.run
加上可执行权限。
安装,先装最大的,然后再装补丁文件
根据图中进行选择
Do you accept the previously read EULA?
accept/decline/quit: accept
Install NVIDIA Accelerated Graphics Driver for Linux-x86_64 410.48?
(y)es/(n)o/(q)uit: no
Install the CUDA 10.0 Toolkit?
(y)es/(n)o/(q)uit: yes
Enter Toolkit Location
[ default is /usr/local/cuda-10.0 ]:
Do you want to install a symbolic link at /usr/local/cuda?
(y)es/(n)o/(q)uit: yes
Install the CUDA 10.0 Samples?
(y)es/(n)o/(q)uit: yes
Enter CUDA Samples Location
[ default is /home/songsl ]:
Installing the CUDA Toolkit in /usr/local/cuda-10.0 ...
等到安装结束即可.
安装另一个补丁包
测试:
进入例子中随便一个,比如NVIDIA_CUDA-10.0_Samples/1_Utilities/bandwidthTest
make
./bandwidthTest
出现如图结果则pass
cuda到此安装结束。
二:Cudnn安装
官网下载: https://developer.nvidia.com/cudnn
下载cudnn需要注册账号登录。
我选择的deb安装,
Runtime 和 Developer 版本的区别是:
developer library 包含了在Ubuntu系统上开发深度学习时所需的cuDNN头文件,如果你不需要开发编译任何深度学习程序,而只是将其用于运行某些深度学习应用,那么只下载“runtime library”就足够了。
我是把3个文件都下载下来了,分别安装。
cuDNN Runtime Library for Ubuntu16.04 (Deb)
cuDNN Developer Library for Ubuntu16.04 (Deb)
cuDNN Code Samples and User Guide for Ubuntu16.04 (Deb)
安装:
sudo dpkg -i libcudnn7_7.6.5.32-1+cuda10.0_amd64.deb
sudo dpkg -i libcudnn7-dev_7.6.5.32-1+cuda10.0_amd64.deb
sudo dpkg -i libcudnn7-doc_7.6.5.32-1+cuda10.0_amd64.deb
测试:
cat /usr/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
cudnn安装完毕!
谢谢!