看了下其他博主的博客讲的比较详细,但有些地方重点不对,或者有些小地方没有讲挖了点坑。我来写一个我自己的简单的主流程,自己在两台机器上安装都有效,来讲一些其他博主有时候没说清楚的地方作为补充,结合其他博主的一起来看。
1.首先去官网找历史版本,不要胡乱下。放下cuda和cudnn的链接
CUDA Toolkit Archive | NVIDIA Developer
cuDNN Archive | NVIDIA Developer
按照他的指示把包下下来,一步一步跟着输命令行
2.出现权限不够就换root
切换至root用户,输入命令:
su root
,然后输入密码
3.配环境变量 注意sudo vim /etc/profile和sudo vim ~/.bashrc的区别,看看自己到底要装在哪
sudo vim可以,sudo gedit我觉更人性化:sudo gedit /etc/profile
输入
source ~/.bashrc
环境变量立即生效。(否则将在机器重启后才生效)
(其实安装完之后就已经告诉你了环境变量要改到哪)
在文本文本末尾添加环境变量:
export LD_LABRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-10.2/lib64
export PATH=$PATH:/usr/local/cuda-10.2/bin
export CUDA_HOME=$CUDA_HOME:/usr/local/cuda-10.2
4.测试cudnn是否能用
旧版是这个代码:
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
5.如果你没装cuda,你输入nvcc -V,系统会提示:
Command 'nvcc' not found, but can be installed with: sudo apt install nvidia-cuda-toolkit
引用下其他博主的话。因为我也没试过,这种安装方法有点诱人。
不要跟着输入那句,这样会经历一个很长的过程,然后还没安装上cuda,安了一个半小时的文件也不知道到哪里去了,看到网上说可以用下面的命令卸载一下,但是显示没有安装。
sudo apt-get autoremove nvidia-cuda-toolkit
所以,还是按照教程老实安装cuda。