栏目分类:
子分类:
返回
名师互学网用户登录
快速导航关闭
当前搜索
当前分类
子分类
实用工具
热门搜索
名师互学网 > IT > 系统运维 > 运维 > Linux

ubuntu安装cuda和cudnn

Linux 更新时间: 发布时间: IT归档 最新发布 模块sitemap 名妆网 法律咨询 聚返吧 英语巴士网 伯小乐 网商动力

ubuntu安装cuda和cudnn

ubuntu安装cuda(10.0)和cudnn
  • 安装显卡驱动
    • cuda和graphic driver版本对应关系
  • 安装CUDA10.0
    • 下载cuda
    • 安装cuda
    • 查看安装的cuda版本
    • 检测是否安装成功
  • 安装cudnn
    • 下载cudnn
    • 安装cudnn
    • 查询cudnn版本

安装显卡驱动 cuda和graphic driver版本对应关系

一般根据需要,安装指定的cuda版本,因此显卡驱动的版本需要满足最低的版本要求。

查看可安装的驱动版本

ubuntu-drivers devices

自动安装推荐的驱动版本

如果推荐版本满足要求,可以自动安装推荐版本的驱动。

sudo ubuntu-drivers autoinstall

安装后重启

检查驱动是否安装成功

nvidia-smi

显示最高可安装的cuda版本为10.2,当前驱动版本为440。

安装CUDA10.0 下载cuda

参考cuda安装方法,进入cuda官方下载链接下载安装即可;

安装cuda

由于已经有了显卡驱动,不需要进行安装;
cuda安装在默认路径即可;

sudo sh cuda_10.0.130_410.48_linux.run

查看安装的cuda版本
cat /usr/local/cuda-10.0/version.txt

检测是否安装成功
nvcc -V

安装cudnn

cudnn是GPU加速计算深层神经网络的库。

下载cudnn

参考cudnn安装方法,进入cudnn官方下载链接;
选择cuda10对应的cudnn版本,点击cudnn library for linux进行下载,注意需要注册登陆才能下载;

安装cudnn

解压

tar -xvf cudnn-10.0-linux-x64-v7.6.5.32.tgz

依次执行命令

sudo cp cuda/include/cudnn.h /usr/local/cuda-10.0/include
sudo cp cuda/lib64/libcudnn* /usr/local/cuda-10.0/lib64
sudo chmod a+r /usr/local/cuda-10.0/include/cudnn.h /usr/local/cuda-10.0/lib64/libcudnn*
查询cudnn版本
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2


cudnn安装成功。

[1]:Ubuntu18.04 安装Cuda 10.0 和Cudnn
[2]:Ubuntu18.04安装cuda(10.1及10.0)和cudnn
[3]:cuda toolkit documentation
[4]:cuda官方下载链接
[5]:cudnn官方下载链接

转载请注明:文章转载自 www.mshxw.com
本文地址:https://www.mshxw.com/it/288875.html
我们一直用心在做
关于我们 文章归档 网站地图 联系我们

版权所有 (c)2021-2022 MSHXW.COM

ICP备案号:晋ICP备2021003244-6号