Ubuntu16.04装CUDA和cuDNN


前言

前提是显卡驱动已经装好。可以参考Ubuntu16.04安装NVIDIA显卡驱动
笔者要复现的代码环境是ubuntu16.04,Python2.7,后续还要装TensorFlow-GPU。
参考文献:
[1] 理清GPU、CUDA、CUDA Toolkit、cuDNN关系以及下载安装
[2] 显卡驱动与cuda对应关系,cudnn与cuda之间的关系
[3] Tensorflow不同版本要求与CUDA及CUDNN版本对应关系
[4] CUDA Tookit Notes
[5] Ubuntu18.04+RTX 2080Ti+CUDA 10.0 +cuDNN+PyTorch搭建深度学习环境
终端运行nvidia-smi,看图得知显卡的计算能力为460.67,最高可装11.2版本的CUDA。

在这里插入图片描述

因此,根据TensorFlow官网和自己的需求,选择Ubuntu16.04下装CUDA10.1,cuDNN7.6。

在这里插入图片描述

一、CUDA安装

1.查看ubuntu系统位数

终端输入以下命令获取ubuntu系统位数信息待用。

sudo uname --m

评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值