Windows系统下使用TensorFlow2.x完美运行RTX3080和3090教程,附带每个步骤所需软件工具的链接

上万块买的3080和3090在TensorFlow上跑不起来,怎么办?废话不多说,直接进入主题:

1、安装TensorFlow的nightly版本

打开cmd输入:

pip install tf-nightly-gpu

这是TensorFlow2.x的每日更新版本,属于测试版且只能用这个版本,原装的TensorFlow2.x不能使用3080或3090!!!

2、安装cuda11.1

nightly版本调用的是cuda11.1或以上的版本,低于此版本都不可以被调用!
官网下载cuda11.1
或者点击下面链接直接下载:
点击下载cuda11.1

3、安装cudnn11.1

官网下载cudnn
或者点击下面链接直接下载:
cudnn11.1

下载完添加到环境变量这些步骤这里就不说了,不会的可查阅我的另一篇博客:Windows或Linux系统环境下成功配置TensorFlow2.x的GPU版本全过程

按照上述的步骤配置好后,GPU就可以使用了,记录一下我跑通3090的光辉时刻:
在这里插入图片描述
由于太长截不了这么多,所以就把这句话复制过来了:

2021-05-15 20:36:03.650534: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1418] Created device /job:localhost/replica:0/task:0/device:GPU:0 with 18786 MB memory:  -> device: 0, name: GeForce RTX 3090, pci bus id: 0000:41:00.0, compute capability: 8.6

报错解决:

  1. 如果运行代码,成功加载了所有的.dll,但是遇到了import时找不到DLL的错误。
    解决方法请看我的另一篇博客:解决:ImportError: DLL load failed while importing
    _pywrap_tensorflow_internal: 找不到指定的模块
  2. 如果成功加载了所有的.dll,GPU也被读取出来了,但是还是报错:tensorflow.python.framework.errors_impl.UnknownError: Failed to get convolution algorithm.“。中文解释为:无法获取卷积算法,这可能是因为cuDNN初始化失败。
    解决方法请看我的另一篇博客:解决:Failed to get convolution
    algorithm
  3. 如果代码能跑通,但是运行到中途就卡死,这目测是个BUG,我这里提供两个解决方法,亲测超级适用:1、如果代码有@tf.function加速,把@tf.function这行注释掉再运行;2、把TensorFlow的nightly版、cuda和cudnn更新一遍,然后再运行。

如果文章解决您多年未解之谜,请一键三连!

评论 3
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

DK数据工作室

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值