关于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的好像,当时那个总报错,这次这个版本的就没问题。
总结
期间参考了很多博主的教程贴,这里就不一一列出来了,算是记录一下吧,能帮得到别人就更好了