Ubuntu18.04 + cuda 10.2其他 + cudnn 配置
- NVIDIA驱动安装
-
- 直接采用命令行的安装形式
- CUDA 10.x 安装
- Cudnn 安装
- 接下来会安装 anaconda pytorch ,一点点来,目前刚用ubuntu,还有很多不会,也不会截图,所以写的比较烂,之后一点点补充把。有问题,评论一起讨论。
NVIDIA驱动安装
直接采用命令行的安装形式
建议直接采用命令行的安装,这样省事省力。建议开始安装都试一下这种方式。
- 打开软件更新。
我之前关闭了软件设置更新,导致安装不成功。所以建议大家先更新以下系统。 - 安装驱动
这里就不写如何看推荐版本了,直接安装把。// 默认安装最新满足要求的 sudo ubuntu-drivers autoinstall
- 如果安装失败了,可以百度艘以下,apk等安装方式,比较麻烦。
最终安装后,进入 系统设置->详细信息 能看到 图形 栏出现了自己电脑的显卡。
CUDA 10.x 安装
-
安装好驱动后,可以在终端运行 nvidia-smi 最上面有个CUDA Version : xxx ,这个一般是支持CUDA 的最高版本。 一般用9.1, 10.1,10.2 目前挺普遍的。(2021年)
-
进入CUDA官网 link.,这里有一些问题,就是进取默认最新的版本,恨fan,想装想要的版本建议 直接在百度搜索一下。
-
网可以的话,建议 runfile 安装方式
-
等待结果就可以拉。 如果不成功,就还是下载再安装(麻烦)。
-
注意 此时要加入环境变量
#终端打开./bashrcsudo gedit ~/.bashrc
// 安装后运行
在 ./bashrc文件中最后加入: 一定要进目录看看文件名字,我装得10.2 所以文件名字cuda-10.2export PATH=/usr/local/cuda-10.2/bin${PATH:+:${PATH}} export LD_LIBRARY_PATH=/usr/local/cuda-10.2/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}} export CUDA_HOME=/usr/local/cuda-10.2
命令行中输入 : source ~/.bashrc
命令行中输入 : nvcc -V 会有结果。
Cudnn 安装
比较麻烦,但是网上资料很多。
- 进入官网link,登陆(注册以下,多等一会就好了)。
- 下载对应的版本 的(cuDNN Library for Linux (x86))。
- 下载完成后解压并进入文件夹:
// 将CUDNN 复制到CUDA中 sudo cp cuda/include/cudnn.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* sudo cp cuda/include/cudnn_version.h /usr/local/cuda/include/
- 检查是否安装成功
cat /usr/local/cuda/include/cudnn_version.h | grep CUDNN_MAJOR -A 2
有结果就安装成功了。