跳转至

Windows TensorFlow2 安装指南

0. 环境说明

硬件:Ryzen R5 3600 + GTX 2060 Super + 32G

系统:Windows 10 x64 企业版 20H2

1. 前期准备

具体所需的各种应用版本请查看这个表

我这里以安装 tensorflow_gpu-2.4.0 为例。

2. 安装

按上面的顺序,顺序安装。

  • CUDA 的 Samples 目录为:C:\ProgramData\NVIDIA Corporation\CUDA Samples\v11.0
  • 复制 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.0\extras\visual_studio_integration\MSBuildExtensions 中的文件到 C:\Program Files (x86)\Microsoft Visual Studio\2019\Enterprise\MSBuild\Microsoft\VC\v160\BuildCustomizations
  • CUPTI 包含在 CUDA 中。
  • cuDNN 解压后覆盖到目录: C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.0
  • TensorRT 6.0 不安装。

安装 TensorFlow 2

# 需要最新的pip
pip install --upgrade pip

# 当前稳定的CPU和GPU版本
pip install tensorflow

3. 验证

Visual Studio 2019 打开 C:\ProgramData\NVIDIA Corporation\CUDA Samples\v11.0\Samples_vs2019.sln

编译,一共 176 个项目,编译成功 173 个,错误 3 个可以不用管。

simpleMPI:mpi.h 文件不存在。Microsoft MPI 下载页面

simpleVulkan:vulkan/vulkan.h 文件不存在。VulkanSDK 下载页面

segmentationTreeThrust: 编译器堆空间不足,编译命令错误。

python -c "import tensorflow as tf;print(tf.reduce_sum(tf.random.normal([1000, 1000])))"

成功:如果系统返回了张量,则意味着您已成功安装 TensorFlow。请查看教程开始使用。


创建日期: 2021-05-05 01:00:00
最后更新: 2022-07-30 02:00:00

评论