1、CUDA安装
1.1、先卸载系统中旧CUDA版本,命令如下:
sudo /usr/local/cuda-9.0/bin/uninstall_cuda_9.0.pl
1.2、卸载之后,会发现 /usr/local/cuda-9.0目录下仍然有文件存在,将CUDA9.0版本文件删除干净, 命令如下:
rm -r cuda-9.0
1.3、安装cuda10.0
下载cuda10.0的安装文件 (官网下载:https://developer.nvidia.com/cuda-toolkit-archive):
sudo sh cuda_10.0.130_410.48_linux.run
1.4、一直按住回车键,直到服务条款为100%时,执行分别如下命令:
accept
n (注:不安装驱动)
y
y
y
1.5、安装完成后,设置环境变量。
打开root目录下的.bashrc文件(vi ./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
1.6、检查CUDA10.0是否安装成功, 命令如下:
cat /usr/local/cuda/version.txt
2、cudnn安装
2.1、下载所需要的cudnn版本(https://developer.nvidia.com/rdp/cudnn-archive)本人下载的是cudnn7.4.2版本
解压文件(solitairetheme8文件,解压命令详见博客:https://blog.csdn.net/qq_28057379/article/details/106015487),
并在解压出的cuda文件夹中打开终端,执行如下命令:
sudo cp include/cudnn.h /usr/local/cuda/include/
sudo cp lib64/lib* /usr/local/cuda/lib64/
2.2、建立软链接
进入/usr/local/cuda/lib64目录下,执行以下指令:
sudo chmod +r libcudnn.so.7.4.2
sudo ln -sf libcudnn.so.7.4.2 libcudnn.so.7
sudo ln -sf libcudnn.so.7 libcudnn.so
sudo ldconfig
2.3、检查cudnn是否安装成功, 命令如下:
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
如果出现如下字样,说明安装成功
#define CUDNN_MAJOR 7
#define CUDNN_MINOR 4
#define CUDNN_PATCHLEVEL 2
--
#define CUDNN_VERSION (CUDNN_MAJOR * 1000 + CUDNN_MINOR * 100 + CUDNN_PATCHLEVEL)
-----------------------------------over--------------------------------------------------
码字不易,如果感觉对您有帮助,就打个赏吧,后续会有更多有价值博客献给大家,星星之火一起燎原