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()