python3.7对应的tensorflow版本_ubuntu18.04 安装cuda、cudnn、tensorflow和pytorch其实很简单...

本文详细介绍了在Ubuntu18.04上安装NVIDIA驱动、CUDA、cuDNN以及TensorFlow-GPU和PyTorch的步骤,包括查看和选择兼容的版本,以及验证安装成功的操作。
摘要由CSDN通过智能技术生成

目录

一. 安装NVIDIA驱动

1. 首先,查看NVIDIA驱动是否已安装

2.安装NVIDIA驱动

二.安装cuda

1. 查看NVIDIA版本对应的cuda版本

2. 选择CUDA版本(上限原则)

3. 安装cuda

4. 安装cndnn

三.安装tensorflow、pytorch深度学习框架

1. 安装tensorflow-gpu

2. 安装pytorch

四.参考

一. 安装NVIDIA驱动

1. 首先,查看NVIDIA驱动是否已安装

在终端输入以下命令,如果显示以下结果,说明NVIDIA驱动已经正常安装,否则,需要重新安装;

nvidia-smi

b3ceb9ddf4ac3860f2b1a9880bc5460f.png

ccd77f065a6db013c090ef59794b09e6.gif

2.安装NVIDIA驱动

方法一:使用Ubuntu自带的镜像源安装

第一步:先在软件中搜索“软件和更新”,然后选择一个可用的镜像源,我选择的是清华的镜像源。我的选择如下:

1eb1a98a03bcaaab666b7e8f47ef66b0.png

ccd77f065a6db013c090ef59794b09e6.gif

第二步:选择源之后,还是在软件更新中,选择附加驱动,然后选择合适的驱动,应用更改就好,我本机选择的是nvidia-driver-455的驱动。

7605fa9484b2e4964eed7d3ff40f7064.png

ccd77f065a6db013c090ef59794b09e6.gif

第三步:完成安装后,在终端输入 nvidia-smi,如果可以显示以下结果,恭喜你,安装NVIDIA驱动成功啦。

e17bfa801ca1665c762ceb798eb5d3dd.png

ccd77f065a6db013c090ef59794b09e6.gif

方法二:下载NVIDIA官方驱动

第一步: 进入NVIDIA官方下载与之匹配的驱动: https://www.nvidia.cn/geforce/drivers/ ,下图是我本机驱动匹配的信息。点击“开始搜索”,进入NVIDIA驱动下载界面。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值