TensorFlow

[TensorFlow]4.最後關卡 安裝gpu版tensorflow

因為我們pip沒有安裝(你可能已經有裝了拉,我是乾淨系統,所以預設是沒有的)

sudo apt install python-pip

裝好後,在下指令安裝GPU版的tensorflow


pip install tensorflow-gpu==1.4

 

最後測試

$ python

>>> import tensorflow as tf
>>> hello = tf.constant(‘Hello, TensorFlow!’)
>>> sess = tf.Session()
>>> print(sess.run(hello))
Hello, TensorFlow!
>>> a = tf.constant(10)
>>> b = tf.constant(32)
>>> print(sess.run(a + b))
42
>>>

Be the First to comment.

Leave a Comment

發佈留言必須填寫的電子郵件地址不會公開。 必填欄位標示為 *

(若看不到驗證碼,請重新整理網頁。)