大道至简,知易行难
广阔天地,大有作为

Win10下TensorFlow with GPU搭建

主要参考官网的两个链接:
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失败

CUDA版本不到导致import tensorflow失败

 

首先更新下pip:

更新pip

更新pip

然后安装virtualenv:

安装virtualenv

安装virtualenv

新建一个env,切换到其对应的目录中,然后将其激活,再安装TensorFlow:

安装TensorFlow GPU

安装TensorFlow GPU

安装成功:

成功安装了TensorFlow

成功安装了TensorFlow

然后简单地测试(创建session时耗时问题是一个已知问题https://github.com/tensorflow/tensorflow/issues/18652):

tensorflow简单测试(这里是一个BUG)

tensorflow简单测试(这里是一个BUG)

转载时请保留出处,违法转载追究到底:进城务工人员小梅 » Win10下TensorFlow with GPU搭建

分享到:更多 ()

评论 抢沙发

  • 昵称 (必填)
  • 邮箱 (必填)
  • 网址