ubuntu 安装 cuda cudnn

下载 cuda和 cudnn

  1. 了解自己机器的nvidia 显卡驱动支持什么版本的cuda 最新的信息参照官网的文档, 下图是我在写这篇博客的最新配置,第一列是cuda 版本,第二列是 linux 驱动版本, 第三列是windows 驱动版本。
    在这里插入图片描述

  2. 官网下载 你的机器需要的cuda 版本,下图是官网截图的一部分
    在这里插入图片描述

  3. 根据自己的操作系统类型,操作系统版本去选择合适的安装包,这里我的是linux->ubuntu->18.04->runfile
    在这里插入图片描述

  4. 长时间等待cuda的下载, 下载成功后会有一个cuda*.run 的文件,这个就是我们需要安装的文件了

  5. 选择需要的cudnn版本,同样的,cudnn 要和nvidia 显卡驱动以及cuda 版本要兼容,具体的信息看官网链接
    下图是官网截图的最新版本的support matrix,进入官网可以查看全部。
    在这里插入图片描述

  6. 下载对应的cudnn 版本,官网链接 需要登录!!!!

安装cuda

  1. 保证安装的用户有root 权限
  2. bash cuda***.run
  3. 根据提示安装就可以了,这里注意一点,cuda 本身自带显卡驱动,在选择的时候一定要把这一项取消,否则它会覆盖掉原来的显卡驱动
  4. 安装完成,系统默认安装在了/usr/local 里,你会发现这里面多了 cudacudaX.X 文件夹,后者是真正的安装结果,cuda 只是链接它的。
  5. 配置环境变量:
    vim /etc/bash.bashrc
    最后一行加入export PATH=${PATH}:/usr/local/cuda/bin
    再加入一行export LD_LIBRARY_PATH=${LD_LIBRARY_PATH}:/usr/local/cuda/lib64
    source /etc/bash.bashrc 更新环境变量
    6.到这里,cuda就完成安装了

安装 cudnn

  1. 解压刚才下载的cudnn 压缩包
    tar -xzvf cudnn-x.x-linux-x64-v8.x.x.x.tgz 会生成一个cuda 文件夹
  2. 复制文件
$ sudo cp cuda/include/cudnn*.h /usr/local/cuda/include  # root 用户去掉sudo
$ sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64    
  1. 更改权限
    $ sudo chmod a+r /usr/local/cuda/include/cudnn*.h /usr/local/cuda/lib64/libcudnn*

安装完成

nvcc -V 查看是否成功

版本不一样?

nvidia-smi 显示的cuda 版本和nvcc -V显示的cuda的版本不一样,不用担心,跑程序的时候用的是nvcc -V 对应的cuda

  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值