Tensorflow简介及GPU环境配置(win10)

Tensorflow是Google开源的深度学习框架，用于训练神经网络模型

Google的神经网络可视化工具http://playground.tensorflow.org/

python、cuda、cudnn、tensorflow版本对应关系参考https://github.com/fo40225/tensorflow-windows-wheel

 

win10(python3.7.4)、TensorFlow1.13.1、CUDA10.0、CUDNN-10.0 v7.6.5.32

nvcc -V

下载CUDA：https://developer.nvidia.com/nvidia-developer-program

下载CUDNN：https://developer.nvidia.com/cudnn

 

在用户变量的PATH中添加以下变量：

CUDA\lib\x64的绝对路径

CUDA\bin的绝对路径

 

import numpy as np
import tensorflow as tf


#查看tensorflow版本
print(tf.__version__)

print('GPU', tf.test.is_gpu_available())

# 1.13.1
# True