win10 安装tensorflow GPU版本

安装cuda

官网下载cuda,本文安装的是9.0版本。

下载cudnn

下载对应cuda9.0的cudnn,本文下载的是cudnn7.1,下载解压,将解压到的文件夹里面的bin,lib,include三个文件夹的文件复制到cuda对应的文件夹里。

安装Anaconda3

TensorFlow 只支持Python3以上版本,因此需要安装Anaconda3.

安装 TensorFlow

  1. 首先,为TensorFlow创建一个虚拟环境:
conda create --name tensorflow-gpu python=3.5

如果报LOCKERROR:It looks like conda is already doing something.执行

conda clean --lock

2)激活/进入虚拟环境

activate tensorflow-gpu

3)安装TensorFlow with GPU support

pip install  --upgrade tensorflow-gpu

测试TensorFlow是否安装完成

import tensorflow as tf
hello =  tf.constant('Hello, TensorFlow!')
sess = tf.Session()
print(sess.run(hello))
#输出:Hello, TensorFlow!
a = tf.constant(10)
b = tf.constant(20)
print(sess.run(a+b))
#输出:30
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容