Windows 平台安装 TensorFlow GPU

2021-02-06 19:17

阅读：607

标签：window 解决 win 错误 ++ update round 方法 pip

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

没有错误，安装成功。

Windows 平台安装 TensorFlow GPU

标签：window 解决 win 错误 ++ update round 方法 pip

原文地址：https://www.cnblogs.com/petewell/p/11410475.html

文章来自：搜素材网的编程语言模块，转载请注明文章出处。
文章标题：Windows 平台安装 TensorFlow GPU
文章链接：http://soscw.com/index.php/essay/51842.html

亲，登录后才可以留言！