鑒于pip安裝tensorflow容易出現(xiàn)runtime error的問(wèn)題,本文使用anaconda中的conda install安裝tensorflow.?
默認(rèn)cuda和 cudnn已經(jīng)安裝好,本文從安裝Anaconda開(kāi)始。
1. 安裝Anaconda
? ?方法(1)?
????????????到 Anaconda download site下載并按照網(wǎng)頁(yè)上的步驟安裝Anaconda
? ?方法(2)需要先下載
? ? ? ? ? ? ?$ bash Anaconda2-4.3.1-Linux-x86_64.sh #Python 2.7版本
? ? ? ? ? ? ?或者
? ? ? ? ? ? ?$ bash Anaconda3-4.3.1-Linux-x86_64.sh #Python 3.5 版本
? ??在安裝的過(guò)程中,會(huì)詢問(wèn)安裝路徑,按回車即可。之后會(huì)詢問(wèn)是否將Anaconda安裝路徑加入到環(huán)境變量(.bashrc)中,輸入yes,這樣以后在終端中輸入python即可直接進(jìn)入Anaconda的Python版本(如果你的系統(tǒng)中之前安裝過(guò)Python,自行選擇yes or no)。安裝成功后,會(huì)有當(dāng)前用戶根目錄下生成一個(gè)anaconda2的文件夾,里面就是安裝好的內(nèi)容
NOTE:安裝完成后需重啟終端conda才能生效
查詢安裝信息
$ conda info
查詢當(dāng)前已經(jīng)安裝的庫(kù)
$ condalist
安裝庫(kù)(***代表庫(kù)名稱)
$ conda install
更新庫(kù)
$ conda update
2. 安裝tensorflow(conda install安裝)
? ? ? ? ?到anaonda cloud tensorflow里面查資源和安裝命令
3. 驗(yàn)證是否安裝成功
python
>>>import tensorflow?as tf
沒(méi)有報(bào)錯(cuò)則安裝成功
4. NOTE:?
指定運(yùn)行時(shí)用的GPU,先查看那個(gè)GPU空閑,然后再指定。
nvidia-smi
import os
os.environ["CUDA_VISIBLE_DEVICES"]='1'