Win10下安装tensorflow(gpu版)
我安装的环境
系统:win10系统
python:3.5
GPU:GeForce GTX1660Ti
CUDA:cuda_9.0.176_win10
CUDNN:cudnn-9.0-windows10-x64-v7
1.安装anaconda
我安装的时anaconda版本是4.2.0,安装包在官网下载,anaconda下载,安装包下载完成后安装,安装过程很简单,一直next即可。
2.安装vs2017(vs2015也可以)
该步很重要,一定要确保安装了vs,若是没有安装vs,直接安装CUDA会导致安装失败,出现很多问题。
vs2017下载
3.安装cuda9.0
CUDA安装过程中会出现各种各样的坑,我也是安装了好多次才成功的。
该部分安装失败可参考博(https://blog.csdn.net/zzpong/article/details/80282814)和(https://blog.csdn.net/xhbspark/article/details/89929310)
4.CUDA9.0安装后测试
1)打开cmd,输入两次cd… ,进入CUDA9.0安装路径下的demo_suite文件夹
2)运行该文件夹下的两个.exe文件
若result=PASS,则表示安装成功,否则安装失败,需要重新安装
5.下载cudnn
这个步骤比较简单,只需要将解压文件中的文件放在CUDA9.0安装路径中的相应文件夹中即可。
6.安装tensorflow-gpu
在下面表中找出对应的tensorflow版本
使用如下命令安装tensorflow-gpu:
pip install --ignore-installed --upgrade tensorflow-gpu -i https://pypi.tuna.tsinghua.edu.cn/simple
7.测试安装是否成功
运行如下程序测试:
import tensorflow as tf
a = tf.random_normal((100, 100))
b = tf.random_normal((100, 500))
c = tf.matmul(a, b)
sess = tf.InteractiveSession()
若运行后结果如红色框中所示,则表示安装成功