2017-07-24 27 views
3

我以前問過是否有可能在CPU上運行張量流與CPU支持。我被告知,這是可能的和基本的代碼來切換我想要使用的設備,但不知道如何獲得在沒有gpu的計算機上工作的初始代碼。例如,我想在裝有NVidia gpu的計算機上訓練,但只能在只有cpu的筆記本電腦上進行編程。我將如何去做這件事?我試圖只是寫代碼,但它甚至可以切換我想要使用的設備之前崩潰。我在Linux上使用Python。如何開發無GPU的張量流與gpu

+0

什麼編程語言?什麼OS? – Neil

+0

@Neil我已經編輯了你所要求的信息。 –

回答

0

這個線程可能會有所幫助:Tensorflow: ImportError: libcusolver.so.8.0: cannot open shared object file: No such file or directory

我試圖導入tensorflow與單向的HPC登錄節點,它不具有GPU的加載tensorflow-GPU。它運作良好。我的筆記本電腦中沒有Nvidia GPU,所以我從未完成安裝過程。但我認爲原因是它無法找到CUDA相關的庫,cuDNN。

但是,你爲什麼不使用CPU版本?正如@Finbarr Timbers所提到的,您仍然可以在帶有GPU的計算機上運行模型。

+0

我想我很想理解一些東西。我是否在我的開發計算機上安裝了cpu版本,沒有gpu的版本,並且在我打算培訓的計算機上安裝gpu版本? –

+0

是的,那正是你應該做的。 – finbarr

0

你得到了什麼錯誤?很有可能在GPU上訓練,但在CPU上進行開發 - 很多人都這樣做,包括我自己。事實上,如果possible的Tensorflow會自動將您的代碼放在GPU上。

如果將以下代碼添加到您的模型,你可以看到正在使用的設備:

# Creates a session with log_device_placement set to True. 
sess = tf.Session(config=tf.ConfigProto(log_device_placement=True)) 

當你與一個GPU的計算機上運行模型這應該改變。

+0

我已經通過thonny安裝tensorflow-GPU和使用驗證碼tensorflow提供,並且我收到以下錯誤: https://pastebin.com/5RxtH5Gx 我得到同樣的錯誤只用代碼: 進口tensorflow爲tf –

+0

我嘗試了您提供的代碼,並且得到相同的錯誤。 –

+0

錯誤是什麼? – finbarr