鲁培军
如何看linux系统中有没有安装cuda,Linux系统 课堂教学管理CUDA10.2+CUDNN安装教程
这⾥主要针对驱动已经安装好了的环境，如何安装合适的显卡驱动这⾥不多赘述，本⽂演⽰的系统为Ubuntu18.04，但理论上其他Linux 发⾏版操作类似。
驱动适配
通过命令⾏输⼊nvidia-smi查看⾃⼰的显卡驱动版本以及⽀持的最⼤CUDA版本，下图第⼀⾏就显⽰了这些信息，可以看到，最⼤⽀持CCUDA10.2，更⾼版本的CUDA需要升级驱动程序。 CUDA的安装可以访问官⽹开发者⼯具中到，链接给出，此时显⽰的结果如下，这是最新的CUDA11，点击右侧的红框选择历史版本。 到CUDA10.2，按照需求进⾏选择，然后安装指⽰使⽤wget或者浏览器下载runfile⽂件。
CUDNN的下载也类似，链接也给出，这个下载需要注册开发者账号(简单⼏步注册登录即可)，然后如下图选择合适的CUDA版本对应的CUDNN并选择CUDNN Library for Linux。这⾥注意，推荐的是CUDNN8.0.2，事实上，还有⼀个7.6.5更合适⼀些，点开Archived菜单查，后者更稳定。
安装
CUDA安装
下⾯的步骤先看完再做。
⾸先，通过下⾯的命令关闭图形界⾯管理器，如果你的图形管理器使⽤的是gdm3那么将命令中的lightdm换成gdm3即可。
南昌七城会sudo service lightdm stop
此时，桌⾯环境会关闭，通过快捷键Ctrl+Alt+F1进⼊命令⾏界⾯，输⼊⽤户名和密码登录，然后到之前下载的cuda⽂件(切换到该⽂件所在⽬录下)，执⾏下⾯的命令。
sudo sh cuda_*.run
此时会进⼊安装，下⾯的过程不便截图，描述⼀下：⾸先，会询问显⽰管理器仍有开启，是否继续安装，这⾥选择continue；然后，会列出⼀个列表要求选择想要安装的内容，这⾥将第⼀个驱动安装的部分回车⼀下将那个x取消，只安装cuda tool kit即可；之后，⽅向键选择下⾯的Install进⾏安装，最后安装成功后会有⼀个提⽰。
这时候cuda是安装好了，但是环境变量不到cuda程序，⾃⼰⼿动修改环境变量⽂件，通过vim或者gedit修改，对应命令分别为vim
~/.bashrc和gedit ~/.bashrc，将下⾯的内容添加到⽂件最后，再通过source ~/.bashrc更新环境变量。
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda/lib64
export PATH=$PATH:/usr/local/cuda/bin
export CUDA_HOME=$CUDA_HOME:/usr/local/cuda
此时，输⼊nvcc -V检查安装是否成功，如下图则表⽰安装成功。
CUDNN安装
切换到cudnn⽂件所在⽬录，通过tar -xzvf cudnn-10.2-linux-x64-v7.6.命令解压⽂件，会得到⼀个cuda⽂件夹，逐⼀执⾏下⾯的命令进⾏cudnn的安装。
李嘉图等价
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*
完成后，通过下⾯的命令查看安装情况，如果结果如下图逐⾏显⽰版本号，则安装成功。
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
⾄此，在Ubuntu18.04上CUDA和CUDNN安装完成。
sa2
训练测试
按照官⽅教程，在虚拟环境中安装Pytorch1.5并进⾏GPU训练测试，结果如下，成功使⽤CUDA训练。
>商标数据库
豫公网安备41160202000603
站长QQ:729038198
关于我们
投诉建议