关于RTX30系显卡安装tensorflow-gpu的问题

关于RTX30系显卡安装tensorflow-gpu的问题

之前按照网上的教程安装nightly版的tensorflow-gpu,但是总报错误,昨天重新配置了一个环境终于不报错了,跟大家分享一下。

环境

Win10+Python3.6+CUDA11.1+Cudnn8.0.4

安装

我首先用conda install pytorch torchvision cudatoolkit=11 -c pytorch-nightly来安装了最新的pytorch
然后在https://pypi.org/下载的tf_nightly_gpu-2.5.0.dev20201217-cp36-cp36m-win_amd64.whl
之前也是在这个网站里下载的dev20201205的好像,当时那个总报错,这次这个版本的就没问题。

总结

期间参考了很多博主的教程贴,这里就不一一列出来了,算是记录一下吧,能帮得到别人就更好了

评论 6
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值