安装cuda
-
从https://developer.nvidia.com/cuda-toolkit-archive下载对应版本的cuda
-
进入到含有安装包的目录下,执行
sudo sh cuda_10.2.89_440.33.01_linux.run
-
等待一会,出来下面的界面,输入
accept
4.cuda安装包
是自带显卡驱动的,所以这一步按空格去掉安装显卡驱动的选项,然后选择install
-
安装完后,应该显示下面的界面
-
需要我们配置环境变量,接下来打开终端输入
gedit ~/.bashrc
,在文件最后输入export PATH=/usr/local/cuda-10.1/bin${PATH:+:${PATH}} export LD_LIBRARY_PATH=/usr/local/cuda-10.1/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
注意
:local后面的名字要安装自己下载的cuda版本,我这里是10.1,所以是cuda-10.1 -
更新环境变量
source ~/.bashrc
-
验证,打开终端输入
nvcc -V
,显示如下则证明cuda安装成功nvcc: NVIDIA (R) Cuda compiler driver Copyright (c) 2005-2019 NVIDIA Corporation Built on Fri_Feb__8_19:08:17_PST_2019 Cuda compilation tools, release 10.1, V10.1.105
安装cudnn
-
从https://developer.nvidia.com/cudnn下载相应版本的cudnn,需要登陆。
-
下载解压
tar -xvf cudnn-10.2-linux-x64-v7.6.5.32.tgz
-
将
cuda/include/cudnn.h
文件复制到usr/local/cuda/include
文件夹;
将cuda/lib64/
下所有文件复制到/usr/local/cuda/lib64
文件夹中,并添加读取权限:sudo cp ./cuda/lib64/* /usr/local/cuda/lib64 sudo cp ./cuda/include/cudnn.h /usr/local/cuda/include sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*
-
查看是否安装成功
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
或者
cat /usr/local/cuda/include/cudnn_version.h | grep CUDNN_MAJOR -A 2
根据cudnn版本不同,使用的命令不同。
显示如下:
#define CUDNN_MAJOR 8 #define CUDNN_MINOR 0 #define CUDNN_PATCHLEVEL 4 -- #define CUDNN_VERSION (CUDNN_MAJOR * 1000 + CUDNN_MINOR * 100 + CUDNN_PATCHLEVEL) #endif /* CUDNN_VERSION_H */
如果找不到的话,查看
cudnn_version.h
文件是否没有移动到/usr/local/cuda/include/
下,新版本好像单独分离了,移动过去就可以了sudo cp ./cuda/include/cudnn_version.h /usr/local/cuda/include