安装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的版本对应关系: 这里
共有 0 条评论