win10 GTX1060 tensorflow-gpu环境部署

win10+GTX1060+cuda9+vs2017+cudnn7+tensorflow-gpu1.5

为了跑神经网络模型新购了一张GT1060,搭建环境碰到一些坑,记下来分享给大家
安装顺序(很重要)!!!
显卡驱动-vs2017->cuda驱动->cudann(复制)
annaconda3.5,(python3.6) tensorflow1.5这两个顺序没要求
各个包的下载地址:
链接:https://pan.baidu.com/s/1i7-5Vnm5AbwFawzPdDVWpQ
提取码:bz2t
1.先安装显卡驱动i-388版本 GeForce_Experience_v3.18.0.102
2.安装vs2017,选择vc++
win10 GTX1060 tensorflow-gpu环境部署_第1张图片
3.安装cuda_9.0.176_win10.exe
https://developer.nvidia.com/cuda-toolkit-archivewin10 GTX1060 tensorflow-gpu环境部署_第2张图片
4.安装cudnn
cudnn-9.0-windows10-x64-v7.zip
将下载的cudnn包解压,将解压后的包里的3个文件copy到CUDA9.0安装路径的对应相同的文件夹里面,比如把%cudnn_pkg_dir%\bin\cudnn64_7.dll 拷贝到%CUDA_install_dir%\CUDA\v9.0\bin下,CUDA默认的安装路径是C:\Program Files\NVIDIA GPU Computing Toolkit 。cudnn可以看作是CUDA的补丁。

5.测试
win10 GTX1060 tensorflow-gpu环境部署_第3张图片

win10 GTX1060 tensorflow-gpu环境部署_第4张图片Result=Pass表示ok

最后安装了tf1.9,1.8都因为提示? MakeShape@TensorShapeUtils@tensorflow@@sa?AVStatus@2@V? S p a n @ Span@ Span@$CBH@absl@@PEAVTensorShape@2@@z(in DLL C:\Users\willy_sung\AppData\Local\Continuum\anaconda3\envs\venv\lib\site-packages\tensorflow\contrib\coder\python\ops_coder_ops.so)"
最后测试了tensorflow-gpu-1.5版本ok
测试代码:
from future import print_function
import tensorflow as tf
import os
import numpy as np
from tensorflow.contrib.keras import keras
import datetime
os.environ[‘TF_CPP_MIN_LOG_LEVEL’]=‘3’
with tf.Session(config=tf.ConfigProto(allow_soft_placement=True, log_device_placement=False)) as sess:
a = tf.constant(1)
b = tf.constant(3)
c = a + b
print(‘结果是:%d\n 值为:%d’ % (sess.run©, sess.run©))

你可能感兴趣的:(开发环境)