如何安装自己想要的版本GPU TensorFlow版本
1.根据自身需求,确定要安装的TensorFlow版本
一个比较简单的方法,找一个普通的服务器环境,首先更新pip,然后执行如下命令,就能列出所有的TensorFlow的版本,接着选择自己要装的版本。
pip3 install tensorflow-gpu==
2.根据要安装的TensorFlow,确定适配的CUDA
这里以tensorflow-gpu==2.2.0为例,可知CUDA 需要10.1,cuDNN需要7.6
https://tensorflow.google.cn/install/source?hl=zh-c