ubuntu16.04+GTX1650 安装cuda,驱动会自动安装!+安装cudnn

补充一句

只是做深度学习跑模型的话,装完驱动后就可以了,不需要额外按照NVIDIA官网的教程配置CUDA和CUDNN!!
我们的代码实际用到的是python环境下的包。比如我用anaconda配置环境,直接在python环境下安装pytorch cudatoolkit就好了。
目前网上的教程提到配环境,几乎都是装完驱动再装cuda和cudnn,实际上后面两个步骤对AI研究者来说,并不必要!
直接在anaconda里配置环境就好了!

按照官网教程安装pytoch cudatoolkit等,cudnn 已经内置在pytorch 中。

conda list
cudatoolkit               11.3.1               h2bc3f7f_2    defaults
pytorch                   1.10.0          py3.7_cuda11.3_cudnn8.2.0_0    file:///home/xiu/Downloads/packages

使用 PyTorch 查看 CUDA 和 cuDNN 版本:

import torch
print(torch.__version__)

print(torch.version.cuda)
print(torch.backends.cudnn.version())

conda安装pytorch后没有找到cudnn

安装CUDA,自动安装驱动

网上一大堆通过runfile来安装的范例,但个人尝试后all not work,最后通过官网教程得以成功安装,并意外发现直接安装cuda是会自动安装显卡驱动的!!(deb文件会自动安装,网上其他案例里选择runflie的话好像是可以选择是否要安装驱动)
以下是一些成功安装的经历。

直接从官网找对应的CUDA Toolkit,点击进行下载。
在这里插入图片描述
根据自己电脑配置对应选择,安装类型我选择了deb(local)。按照官方给出的指令进行下载。
在这里插入图片描述

按照官方给出的指令执行后,在安装过程中自动安装了对应的显卡驱动!!!早知道就不折腾上网找驱动的安装教程了,直接安装cuda,自动安装驱动的方式不好吗!!

安装NVIDIA驱动教程,试了一圈都没用:
https://tensorflow-notes.readthedocs.io/zh_CN/latest/nv-driver.html
Ubuntu16.04下安装cuda和cudnn的三种方法(亲测全部有效

安装完成后reboot,独显可以正常显示了!!
在这里插入图片描述
在这里插入图片描述
内心沸腾!!!

nvcc -V不显示问题

但此时输入 nvcc -V显示没用安装,

xiu@xiu:~$ nvcc -V
The program 'nvcc' is currently not installed. You can install it by typing:
sudo apt install nvidia-cuda-toolkit

网上搜索,原来是环境变量没有设置,
打开~/.bashrc ,添加环境变量export PATH=$PATH:/usr/local/cuda/bin
重新执行nvcc -V,输出显示cuda版本为11.1.
(参考:解决nvcc找不到的问题)

xiu@xiu:~$ nvcc -V
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2020 NVIDIA Corporation
Built on Mon_Oct_12_20:09:46_PDT_2020
Cuda compilation tools, release 11.1, V11.1.105
Build cuda_11.1.TC455_06.29190527_0

大功告成!

安装cudnn

参考:
cuDNN的安装(版本选择, Runtime 还是 Developer)

Ubuntu16.04下安装cuda和cudnn的三种方法(亲测全部有效

官方教程

官方下载地址:https://developer.nvidia.com/rdp/cudnn-archive

文件选择cuDNN Library for Linux 即可。
在这里插入图片描述

安装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 /usr/local/cuda/lib64/libcudnn*

查看cudnn版本:

#给笔记本装的cuda11.1 cudnn版本记录在cudnn_version.h里
cat /usr/local/cuda/include/cudnn_version.h | grep CUDNN_MAJOR -A 2
#or 给台式机装的cuda9.2 cudnn版本记录在在cudnn.h里
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
  • 5
    点赞
  • 18
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

xiu_cs

你的鼓励是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值