Windows10安装tensorflow

发表于

Requirements

  1. Python3.X
  2. Windows10

安装tensorflow

1
2
3
4
5
# GPU版本
pip3 install --upgrade tensorflow-gpu

# CPU版本
pip3 install --upgrade tensorflow

注：如果显卡是英伟达支持CUDA的独显就安装GPU版本

查询是否支持CUDA

我安装GPU版本成功但是没效果，只能安装CPU版本了

安装CUDA和cuDNN

如果是CPU版本就跳过这个步骤

CUDA

下载地址

可选在线安装版本和离线版本

安装完成后CMD输入nvcc - ，出现版本信息即可

cuDNN

下载地址

下载完成是一个压缩包，解压到任意目录即可。

然后把目录下的bin目录路径添加到环境变量的path里

测试

随便写个程序测试一下

1
2
3
4
import tensorflow as tf
hello = tf.constant('Hello, TensorFlow!')
sess = tf.Session()
print(sess.run(hello))