Tensorflow2 - gpu 使用 pip 在线安装

• 安装之前请确定 Tensorflow2 - gpu 版本号对应的 python、cuda、cudann 的版本号：官方测试的构建配置https://tensorflow.google.cn/install/source_windows#gpu
• 本文提供主要是提供 pip 和 conda 安装方法，全程在线下载安装，请保持稳定的网络连接。

---

1. 在 cmd 中进入指定的虚拟环境（conda activate xxxx（提前创建好的环境名字））；
2. 确定 conda 的镜像是否已替换成清华源或豆瓣源或阿里源（替换成国内任意源即可，推荐清华源）；
3. 依次按照一下步骤安装（示例），安装过程中会下载相关包，耐心等待即可：

pip install tensorflow-gpu==2.3.0 -i https://pypi.tuna.tsinghua.edu.cn/simple

conda install cudatoolkit=10.1 cudnn=7.6.5

---

• 安装完成后在 cmd 中依次输入一下代码测试：

 

conda activate xxxx（环境名）


python


import tensorflow as tf


print(tf.__version__)

最后返回 tensorflow 的版本号表示安装成功。