ubuntu安装cudo10.0配置cudnn7.6
1、先卸载掉原先安装的cuda10.0或者cuda10.1(如果未安装,直接第二步)
2、安装cuda
sudo sh cuda_9.0.176_384.81_linux.run
会出现很长的提示信息,按下“F2”键跳过
3、设置环境变量
geidt ~/.bashrc
将以下内容复制进去:
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-10.0/lib64
export PATH=$PATH:/usr/local/cuda-10.0/bin
export CUDA_HOME=$CUDA_HOME:/usr/local/cuda-10.0
终端运行:source ~/.bashrc
检查:nvcc --version,如果显示下面的文子就说明安装成功了。
4、安装cudnn
解压下载的文件,可以看到cuda文件夹,在当前目录打开终端,执行如下命令:
sudo cp cuda/include/cudnn.h /usr/local/cuda-10.0/include/
sudo cp cuda/lib64/libcudnn* /usr/local/cuda-10.0/lib64/
sudo chmod a+r /usr/local/cuda-10.0/include/cudnn.h /usr/local/cuda-10.0/lib64/libcudnn*
5、查看cudn版本
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2