1、根据自己的系统在官网下载cuda (选择runfile(local))
https://developer.nvidia.com/cuda-downloads
2、进入下载目录,并执行
sh cuda_XXXX_linux.run
注意:1.在协议中选择同意EULA(accept)
2.不安装driver installation (no)
3.选择cuda安装目录,如/home/yourname/cuda,以及cudasamples的目录
3、下载cudnn,下载地址 https://developer.nvidia.com/cudnn
4、解压cudnn
tar -xzvf cudnn-9.0-linux-x64-v7.tgz
5、拷贝cudnn部分文件到cuda
cp cuda/include/cudnn.h cuda9/include/ cp cuda /lib64/libcudnn* cuda9/lib64 chmod a+r cuda/include/cudnn/h cuda/lib64/libcudnn*
6、修改个人用户的环境变量
环境变量文件 ~/.bashrc位于home/yourname/~/.bashrc(用vim ~/.bashrc编辑)
“export PATH=$HOME/cuda9/bin:$PATH” “export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:$HOME/cuda9/lib64/”
修改之后 source ~/.bashrc使环境变量生效