Windows 平台安装 TensorFlow GPU

Posted 2022-08-27 petewell

原文引用https://www.dazhuanlan.com/2019/08/26/5d63016142d97/

国际惯例，看官方文档。官方推荐使用pip安装，安装 TensorFlow 之前需要安装

CUDA 10.1
cuDNN 7.6
Microsoft Visual C++ 2015 Redistributable Update 3

然后执行

pip install tensorflow-gpu

进行安装。

测试一下安装是否成功：

import tensorflow as tf

报错，提示缺少 DLL 文档。搜了一下是咋回事，大多人解决的方法是安装VC++ 2015 Redistributable Update 3，重装VC++ 2015 Update 3后卸载TensorFlow-GPU再安装，还是报同样的错误。

换用 Conda 安装

在装PyTorch时看到官方安装文档里有用Conda安装的方法，就顺道在TensorFlow上试一下吧。

# 安装 PyTorch
conda install pytorch torchvision cudatoolkit=10.0 -c pytorch
# 安装 TensorFlow
conda install tensorflow-gpu

安装后测试：

import tensorflow as tf
import torch

没有错误，安装成功。