python - tensorflow 在windows 下怎么使用gpu ?
迷茫
迷茫 2017-04-18 10:12:20
0
1
933
  1. 相关环境

  • windows 10

  • python 3.5.2

  • tensorflow-gpu (0.12.1)

  • N卡960M

  • 核显 Intel HD Graphics 4600

  1. 代码

import tensorflow as tf

with tf.Session() as sess:
  with tf.device("/gpu:0"):
    matrix1 = tf.constant([[3., 3.]])
    matrix2 = tf.constant([[2.],[2.]])
    product = tf.matmul(matrix1, matrix2)
    result = sess.run([product])
    print(result)
  1. 错误信息

InvalidArgumentError (see above for traceback): Cannot assign a device to node 'MatMul': Could not satisfy explicit device specification '/device:GPU:0' because no devices matching that specification are registered in this process; available devices: /job:localhost/replica:0/task:0/cpu:0
         [[Node: MatMul = MatMul[T=DT_FLOAT, transpose_a=false, transpose_b=false, _device="/device:GPU:0"](Const, Const_1)]]

2017-01-08 tensorflow 在windows 下使用gpu中文教程

迷茫
迷茫

业精于勤,荒于嬉;行成于思,毁于随。

全員に返信(1)
黄舟

以前のバージョンはdockerでのみ実行できるため、GPUを使用できません。新しいバージョンでは Windows をサポートしていると記載されていますが、これらを正しくインストールしている限り、問題はありません。

Python3.5、TensorFlow、Cuda ツールキット 8.0、cuDNN。

詳しくはこの記事をお読みください http://www.netinstructions.co...

いいねを押す +0
人気のチュートリアル
詳細>
最新のダウンロード
詳細>
ウェブエフェクト
公式サイト
サイト素材
フロントエンドテンプレート