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++
在这里插入图片描述
3.安装cuda_9.0.176_win10.exe
https://developer.nvidia.com/cuda-toolkit-archive在这里插入图片描述
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.测试
在这里插入图片描述

在这里插入图片描述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©))

  • 0
    点赞
  • 4
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值