安装tensorflow-gpu和tensorflow_federated

前言
在安装tensorflow-gpu前要先安装CUDA和cuDNN,具体安装步骤可以见上一篇文章:记录Win10正确安装CUDA和cuDNN的过程(记录一些坑) 。
安装tensorflow-gpu
我电脑上安装的CUDA版本为10.2,cuDNN版本为7.6.5,根据TensorFlow官网的对照表: tensorflow-gpu 2.1.0、2.2.0、2.3.0都可以安装,这里建议安装低版本2.1.0,不过我安装了2.3.0。
进入Anaconda Prompt,输入:
pip install tensorflow-gpu==2.3.0 -i https://pypi.tuna.tsinghua.edu.cn/simple

安装成功!
安装tensorflow_federated
GitHub查看tensorflow和tensorflow_federated的版本对应关系: 这里

安装tensorflow-gpu和tensorflow_federated最先出现在Python成神之路

版权声明:
作者:dingding
链接:https://www.techfm.club/p/17729.html
来源:TechFM
文章版权归作者所有,未经允许请勿转载。

THE END
分享
二维码
< <上一篇
下一篇>>