【环境配置】Ubuntu20.04 安装cuda和cudnn
安装cuda
-
从https://developer.nvidia.com/cuda-toolkit-archive下载对应版本的cuda
-
进入到含有安装包的目录下,执行
sudo sh cuda_10.1.105_418.39_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.1-linux-x64-v8.0.4.30.tgz -
将
cuda/include/cudnn.h文件复制到usr/local/cuda/include文件夹;
将cuda/lib64/下所有文件复制到/usr/local/cuda/lib64文件夹中,并添加读取权限:sudo cp cuda/include/* /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.h里面含有cudnn_version.h,则用下面的命令cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2或者
如果cudnn.h与cudnn_version.h单独分离,则用下面的命令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
更多推荐
所有评论(0)