第一步:安装Anaconda、cuda以及对应的cudNN,其中TensorFlow-GPU对应版本如下图所示:
想看最新的话,可以看官方链接:https://www.tensorflow.org/install/source_windows?hl=zh-cn
第二步:创建对应的虚拟环境,推荐Python3.7
第三步:以我为例:
cuda 11.2 Pyhton 3.7 安装2.6.0
激活虚拟环境,使用清华源镜像
pip install tensorflow-gpu==2.6.0 -i https://pypi.tuna.tsinghua.edu.cn/simple