美文网首页我爱编程
Win7 tensorflow gpu安装

Win7 tensorflow gpu安装

作者: 菜囧爱学习 | 来源:发表于2018-04-13 22:31 被阅读0次

Win7 tensorflow gpu安装

安装环境:

win7 i7 GTX960M

Cuda cuda8.0 (cuda_8.0.61_windows_network)

https://pan.baidu.com/s/1TCxN9UmlRMK3pLPFdQAUDA

Cudnn v5.1(cudnn-8.0-windows7-x64-v5.1)

https://pan.baidu.com/s/1m_NubFo7WzP83TDKAFHzrw

Anaconda3:4.2(Anaconda3-4.2.0-Windows-x86_64,python3.5)

https://pan.baidu.com/s/1PtoX-ELivWaU-3gId0ZzjA

Tensorflow gpu 1.2 cp5

第一步:anaconda3安装 ,记得在安装过程中勾选添加环境变量

第二步:cuda和cudnn安装

1)cuda安装

2)cudnn安装

将解压后的“cuda”文件夹放到计算机某一文件夹,进入“Path”添加环境变量,添加"c:\xxx\cuda\bin"与”C:\Program Files\NVIDIA GPU Computing

Toolkit\CUDA\v8.0\bin”其中xxx为cuda上述某一文件夹

第三步:tensorflow安装

pip install tensorflow_gpu-1.2.1-cp35-cp35m-win_amd64.whl

可以在清华源上下载对应的whl(后缀改为whl)再安装

```pip install --upgrade https://mirrors.tuna.tsinghua.edu.cn/tensorflow/windows/gpu/tensorflow_gpu-1.2.1-cp35-cp35m-win_amd64.whl```

开始安装自动安装tensorflowgpu-1.7报错,原因是1.7版本和cuda8.0不匹配

测试tensorflow-gpu

import tensorflow as tf

a =tf.constant([1.0, 2.0, 3.0, 4.0, 5.0, 6.0], shape=[2, 3], name='a')

b = tf.constant([1.0, 2.0, 3.0, 4.0, 5.0, 6.0], shape=[3, 2], name='b')

c = tf.matmul(a, b)

sess = tf.Session(config=tf.ConfigProto(log_device_placement=True))

print (sess.run(c))

问题1:

在安装tensorflow时出现 “Cannot remove entries from nonexistent file c:\program files\anaconda3\lib\site-packages\easy-install.pth” 的问题。查看原因是因为setuptools版本太低,tensorflow要求29.0.1,当前版本为27.2.0,在更新setuptools版本的时候又找不到easy-install.pth,导致更新失败

运行:pip install --upgrade --ignore-installed setuptools,问题解决!

问题2:

测试时出现gpu无法弹出问题,gpu驱动太新,驱动精灵换了一个2018.03左右的驱动。

相关文章

网友评论

    本文标题:Win7 tensorflow gpu安装

    本文链接:https://www.haomeiwen.com/subject/rgnhkftx.html