关于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_ni
原创
2020-12-22 21:45:21 ·
2415 阅读 ·
6 评论