Windows 平台安装 TensorFlow GPU

原文引用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

没有错误,安装成功。

猜你喜欢

转载自www.cnblogs.com/petewell/p/11410475.html