1. 程式人生 > >TensorFlow-gpu安裝心得,一小時搞定了

TensorFlow-gpu安裝心得,一小時搞定了

搞了三天的TensorFlow-gpu的安裝,終於好了,我稍微總結一下。前兩天一直在我的桌上型電腦上安裝,失敗了好多次,今天晚上我只用了一個小時就把我的xps上也裝上了TensorFlow-gpu。回顧一下安裝步驟吧,希望以後換電腦了也不會忘記。首先,我的桌上型電腦配置是顯示卡GTX 1070,筆記本是GTX970M的,所以安裝python3.6+TensorFlow-gpu1.8+cuda9.0+cudnn7都沒有問題的。
預備材料有以下:1. Anaconda 5.0.1;2. Pycharm(看個人了)3. CUDA9.0+cudnn7有了這些東西,在配置好cudnn7的環境變數(Anaconda自己會新增環境變數的),以及下一步下一步的安裝好Anaconda之後,就可以直接安裝了,進入cmd直接上命令吧。1. conda info--envs(檢視一下conda的環境,最初只有anaconda一個環境,前面加一個小*號)2. conda search --full-name python(檢視一下可安裝的python版本,這裡我選的是3.6版本的)3. conda create --name tensorflow-gpu python=3.6(安裝python3.6版本)4. activate tensorflow-gpu(此時在命令路徑之前會出現一個帶小括號的(activatetensorflow-gpu))5. conda info--envs(檢視一下conda的環境,此時會出現兩個,多出來一個tensorflow-gpu)6. python --version(檢視一下python版本是否安裝好)7. pip install -i https://pypi.tuna.tsinghua.edu.cn/simple/tensorflow-gpu(安裝tensorflow-gpu,從國內的清華的映象走,國外下載的太慢了)8. 進入python編譯環境,輸入import tensorflow as tf,回車9.上一步回車成功了,沒有報錯,萬幸,然後輸入tf.__version__(下面是兩個下劃線),如果顯示出TensorFlow-gpu的版本,證明安裝成功了。三天的時間,終於搞明白這個東西安裝的過程了,中間也遇到了很多問題,比如說先裝了TensorFlow的CPU版本之後,再安裝GPU版本就始終無法載入modular了,很蛋疼。對了,cudnn的路徑一定要新增到環境變數,cuda的安裝一定要小心,反正我第二次裝的時候直接就是下一步下一步的,也不知道為啥第一次裝的時候就那麼難。我之前就是因為這些個小細節沒有做到,結果讓我解除安裝了重灌,重灌了在解除安裝。。。PS:感謝95和小白的大力幫助,萬分感謝,雖然你們不知道這個是我的部落格,哈哈~好好加油開始搞我的DQN吧,A za A za fighting!!!爭取早日出論文早日...咳咳,早日漲工資吧。