win10 GTX1060 tensorflow-gpu环境部署_gtx1060能跑神经网络吗

作者：繁依Fanyi0 | 2024-04-07 18:50:41

踩

gtx1060能跑神经网络吗

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 @$ $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©))

声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/繁依Fanyi0/article/detail/379976