1
我有tensorflow GPU與CUDA一起工作。Tensorflow不使用GPU與jupyter（Windows）
但是當我運行在jupyter筆記本下面的代碼：
import sys
import numpy as np
import tensorflow as tf
from datetime import datetime
with tf.device('/gpu:0'):
a = tf.constant([1.0, 2.0, 3.0, 4.0, 5.0, 6.0], shape=[2, 3], name='a')
b = tf.constant([1.0, 2.0, 3.0, 4.0, 5.0, 6.0], shape=[3, 2], name='b')
c = tf.matmul(a, b)
# Creates a session with allow_soft_placement and log_device_placement set
# to True.
sess = tf.Session(config=tf.ConfigProto(
allow_soft_placement=True, log_device_placement=True))
# Runs the op.
print (sess.run(c))
命令提示符顯示用於CPU，而不是GPU。
最新輪PKG網址你確定你的Jupyter筆記本電腦使用正確的蟒蛇環境（tensorflow-GPU2）？我已經看到它在哪裏使用一些其他的默認環境，並且需要改變它 – Luca
我如何確認它是否使用正確的環境？ – rahulmanuwas
我認爲應該有一個'conda'標籤，至少它會告訴你哪一個是默認的。目前，我會改變這一點，看看是否改變了一些東西。我現在不在開發環境附近，但是您可能想要了解如何啓動筆記本以使用特定的python env。 – Luca