解決Mask RCNN訓練時GPU內存溢出問題

首先自己是個小白對於如何使用GPU跑程序這個學習了一下: (1)使用的是putty,安裝了Anaconda這個IDE環境,用的conda install tensorflow-gpu安裝的GPU版本tf,之前只是安裝了tf並沒有選擇GPU版本。安裝完GPU版本的tf,可以進入python環境導入import tensorflow as tf 進行測試一下,如果沒有報錯即爲成功。 (2)並在自己Tr
相關文章
相關標籤/搜索