看了下其他博主的博客讲的比较详细,但有些地方重点不对,或者有些小地方没有讲挖了点坑。我来写一个我自己的简单的主流程,自己在两台机器上安装都有效,来讲一些其他博主有时候没说清楚的地方作为补充,结合其他博主的一起来看。

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。

Logo

为开发者提供学习成长、分享交流、生态实践、资源工具等服务,帮助开发者快速成长。

更多推荐