windows 10下安装 TensorFlow GPU版本
2019-03-06 21:42:22
•
阅读
打赏
TensorFlow GPU版本需要各种驱动程序和库的支持。此设置仅需要 NVIDIA® GPU 驱动程序。为了简化安装并避免库冲突，建议您使用支持 GPU 的 TensorFlow Docker 映像（仅限 Linux）。
硬件要求
系统支持以下支持 GPU 的设备：（本文使用了GTX 750TI）
- CUDA® 计算能力为 3.5 或更高的 NVIDIA® GPU 卡。请参阅支持 CUDA 的 GPU 卡列表。
软件要求
须在系统中安装以下 NVIDIA® 软件：
- NVIDIA® GPU 驱动程序 - CUDA 9.0 需要 384.x 或更高版本。
- CUDA® 工具包 - TensorFlow 支持 CUDA 9.0。
- CUDA 工具包附带的 CUPTI。
- cuDNN SDK（7.2 及更高版本，需要官网注册下载）
- （可选）NCCL 2.2，可实现多 GPU 支持。
- （可选）TensorRT 4.0，可缩短在某些模型上进行推断的延迟并提高吞吐量。
本文的环境为：Windows 10 + GPU驱动（Driver Version: 391.35 ）+ CUDA9.0 + cuDNN（cudnn-9.0-windows10-x64-v7.3.1.20）
CUDA和CuDNN安装
CUDA是NVIDIA推出的运算平台，CuDNN是专门针对Deep Learning框架设计的一套GPU计算加速方案。
显卡型号支持：https://developer.nvidia.com/cuda-gpus
CUDA下载地址：https://developer.nvidia.com/cuda-toolkit-archive
CuDNN下载地址：https://developer.nvidia.com/rdp/cudnn-download
（下载CuDNN需要注册账号，注册过程比较简单）
安装命令
pip3 install --upgrade tensorflow-gpu
相关推荐
深度学习 -- 损失函数
深度残差网络（Deep Residual Networks (ResNets)）
深度学习 -- 激活函数
神经网络训练 -- 调整学习速率
生成对抗网络（GAN）改进与发展
生成对抗网络（GAN）优点与缺点
生成对抗网络（GAN）的训练
生成对抗网络（GAN）基本原理
生成模型与判别模型