Tensorflow-GPU 环境安装与配置(Windows10)

一、 N卡安装与配置环境变量
  1. 如果还没有安装 Nvidia CUDA 和 cuDNN ,请先点击查看

  2. Tensorflow-gpu 硬件要求:

https://tensorflow.google.cn/install/gpu#hardware_requirements

  1. 小编成功配置的版本及环境:

操作系统:Windows10
系统位数:64 bit
GPU:GeForce GTX 1060 6GB
Nvidia CUDA:cuda_9.0.176_win10
Nvidia cuDNN:cudnn-9.0-windows10-x64-v7.4.2.24
Tensorflow-gpu 版本:1.12.0
Python 环境:3.6.4

二、 安装 Tensorflow-gpu 包
  1. 请选择对应的 Tensorflow-gpu 版本,版本要求如下:

https://tensorflow.google.cn/install/source_windows#gpu

  1. Anaconda 安装方式(推荐方式)(Anaconda 安装指南):
conda install tensorflow-gpu=[版本号]
  1. pip 安装方式:
pip install tensorflow-gpu=[版本号]
  1. 验证安装是否成功:
import tensorflow as tf
hello = tf.constant('hello world')
world = tf.Session()
print(world.run(hello))
  1. 查看 Tensorflow 版本和安装路径:
print('Tensorflow 版本:',tf.__version__)
print('Tensorflow 安装路径:',tf.__path__)

你可能感兴趣的:(Tensorflow-GPU 环境安装与配置(Windows10))