tensorflow_gpu版本安装

tensorflow_gpu版本安装

这是一个非常简洁的安装指南,务必严格按照上述要求检查安装。
https://zhuanlan.zhihu.com/p/183981491

利用anaconda安装 tensorflow_gpu 2.3.0,一开始因为版本一致问题,一直很惧怕安装,最终自己想了最简单的方法,和上面链接方法基本一致,链接介绍了原理!

1.创建Anaconda虚拟环境,同时指定python版本。
打开anaconda prompt 命令行,执行以下命令:

conda create -n tfgpu2.3.0 python=3.7

2.检查nvidia驱动版本,NVIDIA驱动程序需 418.x 或更高版本。
可在命令行中执行查看驱动版本: nvidia-smi
在这里插入图片描述
3.安装依赖库
如上图,可知CUDA版本是11.1。 GPU版本有两个依赖库cuda和cudnn,对于 tensorflow2.3来讲,查阅资料知道:
CUDA的版本需要是 10.1
cudnn版本号需要不小于 7.6
如果直接执行conda install cudatoolkit cudnn,anaconda会自动匹配版本,CUDA的版本是 10.2,我不知道可不可以,有的文章说会产生冲突,我是按照指定版本的方式安装!

3.执行下面命令,cudnn自动匹配的版本应该是7.6.5:

conda install cudatoolkit=10.1 cudnn

4.最后执行tensorflow-gpu安装:

conda install tensorflow-gpu=2.3.0

5.验证安装成功与否,按照图片的命令执行即可!
在这里插入图片描述

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值