win10 + gtx1050 + cuda9.0 + Tensorflow1.8 gpu版本安装

win10 + gtx1060 + cuda9.0 + Tensorflow1.8 gpu版本安装

俺电脑是GTX1050，对应CUDA9.0，

先安装acaconda
再安装cuda9.0，再安装cudnn
再创建虚拟环境
如：

conda create -n your_env_name python=3.6

再安装GPU版本的 tensorflow

pip install tensorflow-gpu==1.8 -i https://pypi.tuna.tsinghua.edu.cn/simple

验证GPU版本的 tensorflow是否安装成功代码

import tensorflow as tf
print('GPU',tf.test.is_gpu_available())
a = tf.constant(2.)
b = tf.constant(4.)
print(a * b)

显示“GPU True”,即代表GPU版本安装成功！

验证代码：

import torch
print torch.cuda.is_available()