Windows TensorFlow2 安装指南¶
0. 环境说明¶
硬件：Ryzen R5 3600 + GTX 2060 Super + 32G
系统：Windows 10 x64 企业版 20H2
1. 前期准备¶
具体所需的各种应用版本请查看这个表
我这里以安装 tensorflow_gpu-2.4.0 为例。
- Python 3.8.x x64 （不要装 3.9.x，不兼容）
- Visual Studio 2019 (需要 MSVC 2019)
- NVIDIA 驱动程序 450.x 或更高版本。
- CUDA 工具包 ：CUDA 11.0 (不要安装其他版本，不兼容)，安装时勾选 CUPTI
- (CUDA 工具包 中包含) CUDA 工具包附带的 CUPTI
- cuDNN SDK 8.0.4 ， cuDNN 下载页面 (不要安装其他版本，不兼容)
- (可选) TensorRT 6.0 ， 下载页面 ，可缩短用某些模型进行推断的延迟时间并提高吞吐量。【Windows 支持较差】
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 tensorflow3. 验证¶
用 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
最后更新: 2022-07-30 02:00:00