cuda
在链接中找到自己想要下载的版本
选择相应的系统及版本,下载runfile
然后运行上图的命令,会有条款,接受即可,注意安装CUDA的时候不要安装驱动(如果已经装过驱动的话)。
然后
sudo gedit ~/.bashrc
在文件最后一行添加:
export PATH=/usr/local/cuda-10.1/bin:/usr/local/cuda-10.1/NsightCompute-2019.1${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-10.1/lib64\
${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
最后,使刚才的配置生效。
source ~/.bashrc
进入/usr/local/cuda-10.1/samples/目录,
sudo make
cd 1_Utilities/deviceQuery
./deviceQuery
出现如下输出,则CUDA安装成功。
cudnn
在链接中下载cudnn
下载完成后解压并进入文件夹:
sudo cp cuda/include/cudnn.h /usr/local/cuda/include/
sudo cp cuda/include/cudnn_version.h /usr/local/cuda/include/
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64/
sudo chmod a+r /usr/local/cuda/include/cudnn.h
sudo chmod a+r /usr/local/cuda/lib64/libcudnn*
在终端查看cudnn版本:
cat /usr/local/cuda/include/cudnn_version.h | grep CUDNN_MAJOR -A 2
出现以下结果