主要参考官网的两个链接:
1、https://tensorflow.google.cn/install/gpu,下载并安装CUDA相关的工具,并注意页面最下方配置PATH;
2、https://tensorflow.google.cn/install/pip
尤其注意选择合适的Python版本,例如当前TensorFlow支持到3.6就不要用3.7,否则实测可能会遇到莫名其妙的问题;另外,CUDA目前只支持9.0(不是9.2,也不是10,在官方链接上明确谢敏了,但GitHub上仍有大量的问题:https://github.com/tensorflow/tensorflow/issues/22794),如果用了9.2或者10.0均会报错:
![CUDA版本不到导致import tensorflow失败](http://www.meilongkui.com/wp-content/uploads/2018/12/tensorflow_dll_load_failed-1024x941.png)
CUDA版本不到导致import tensorflow失败
首先更新下pip:
![更新pip](http://www.meilongkui.com/wp-content/uploads/2018/12/tensorflow_setup_update_pip.png)
更新pip
然后安装virtualenv:
![安装virtualenv](http://www.meilongkui.com/wp-content/uploads/2018/12/tensorflow_setup_install_virtualenv.png)
安装virtualenv
新建一个env,切换到其对应的目录中,然后将其激活,再安装TensorFlow:
![安装TensorFlow GPU](http://www.meilongkui.com/wp-content/uploads/2018/12/tensorflow_setup_install.png)
安装TensorFlow GPU
安装成功:
![成功安装了TensorFlow](http://www.meilongkui.com/wp-content/uploads/2018/12/tensorflow_setup_install_success.png)
成功安装了TensorFlow
然后简单地测试(创建session时耗时问题是一个已知问题https://github.com/tensorflow/tensorflow/issues/18652):
![tensorflow简单测试(这里是一个BUG)](http://www.meilongkui.com/wp-content/uploads/2018/12/tensorflow_test.png)
tensorflow简单测试(这里是一个BUG)
转载时请保留出处,违法转载追究到底:进城务工人员小梅 » Win10下TensorFlow with GPU搭建