Win10+GTX 1080Ti+Anaconda TensorFlow安装

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/Miss_yan/article/details/83007246

实验室新配了电脑,折腾了一整天,总算是把环境配好了

查看了很多blog,都是说GTX1080Ti目前只能使用cuda v8.0(可以参考Keras的中文文档,写的非常详细https://keras-cn.readthedocs.io/en/latest/for_beginners/keras_windows/),当然也有大神尝试过其他版本(win10+1080Ti+cuda9.1+cudnn7+python3.6),但由于本人不敢再踩坑了,所以有兴趣的朋友可以自己试试看啦:)https://blog.csdn.net/u012746060/article/details/81182727

步骤

1、下载cuda v8.0安装,选择自定义,只选第一个就行

2、下载cudnn v6.0并解压

注意:上两步一定要注意版本,不然安了后面还是要卸载重装的

3、将cudnn下面的三个文件复制到cuda安装路径下对应的文件夹中,这里有一个可以测试cuda是否安装成功的方法,参见https://blog.csdn.net/wanglei022/article/details/78659955 使用cmd命令行格式,CD到 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0目录下 找一下 extras\demo_suite 文件夹。里面提供了NVIDIA自带的两个测试.exe文件:deviceQuery.exe 和 bandwidthTest.exe 。 在命令行输入文件名(如bandwidthTest.exe),如果看到输出的最后result = pass 就说明安装成功! 恭喜你:)。

4、创建TensorFlow环境

  conda create -n tensorflow_gpu python=3.5

  activate tesorflow_gpu

  pip install tensorflow-gpu==1.3

 注意,这里tensorflow-gpu要选定1.3版本的

5、测试

import tensorflow,如果没有报错的话,安装成功了

安装期间出现的问题:No module named '_pywrap_tensorflow_internal'

解决方案,注意cudnn和cuda的版本问题,不知道是1080Ti还是win系统的问题,对cudnn和cuda的版本要求很高,真的是差一点都不行。之前尝试过的版本1)cuda v10.0结果失败了2)cuda v8.0+cudnn v7.0失败3)cuda v8.0+cudnn v6.0终于成功了

参考blog

https://blog.csdn.net/he5688/article/details/77542011

https://blog.csdn.net/wanglei022/article/details/78659955

https://keras-cn.readthedocs.io/en/latest/for_beginners/keras_windows/

猜你喜欢

转载自blog.csdn.net/Miss_yan/article/details/83007246