主要参考官网的两个链接:
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均会报错:
首先更新下pip:
然后安装virtualenv:
新建一个env,切换到其对应的目录中,然后将其激活,再安装TensorFlow:
安装成功:
然后简单地测试(创建session时耗时问题是一个已知问题https://github.com/tensorflow/tensorflow/issues/18652):
转载时请保留出处,违法转载追究到底:进城务工人员小梅 » Win10下TensorFlow with GPU搭建