簡介

最近使用tensorflow object detection訓練自己的資料集時，出現了

AttributeError: module 'tensorflow.contrib.data' has no attribute 'parallel_interleave'

主要的原因是tensorflow1.4沒有這個函式，需要tensorflow1.7(不知道1.5,1.6可以嗎)版本。然而直接安裝會出現tensorflow不管用的情況，主要原因是當前的cuda8.0不支援1.7版本。所以需要安裝一個比較新的cuda9.0，因為我已經安裝了nvidia驅動，驅動型號是390.25 (可以參考我之前的部落格)

解除安裝cuda8.0

開啟你的terminal，輸入命令

sudo service lightdm stop
然後ctrl+Alt+F1，輸入賬號，密碼登入
cd /usr/local/cuda-8.0/bin
sudo ./uninstall_cua_8.0.pl

這裡面可能應該會出現一個沒有刪除的情況(應該是cudnn的東西吧)

可以直接將這個資料夾刪除了

sudo rm -rf cuda8.0

安裝cuda9.0

接下來的其實可以參考我的部落格

下載地址

下載安裝，然後我介紹一下流程

cd Dowload/
sudo sh cuda_9.0.176_384.81_linux.run --no-opengl-libs  

然後按照接下來的步驟做,cuda9.0自帶的驅動，可以不用，因為已經安裝好了驅動。

Do you accept the previously read EULA?
accept/decline/quit:accept   

Install NVIDIA Accelerated Graphics Driver for Linux-x86_64 387.26?
(y)es/(n)o/(q)uit: n

Install the CUDA 9.0 Toolkit?
(y)es/(n)o/(q)uit: y

Enter Toolkit Location
 [ default is /usr/local/cuda-9.0 ]: 
直接按enter

Do you want to install a symbolic link at /usr/local/cuda?
(y)es/(n)o/(q)uit: y

Install the CUDA 9.0 Samples?
(y)es/(n)o/(q)uit: y
 

配置環境變數

sudo gedit /etc/profile 

export PATH=/usr/local/cuda-9.0/bin/:$PATH  
export LD_LIBRARY_PATH=/usr/local/cuda-9.0/lib64/ 

測試

 #編譯並測試裝置 deviceQuery：  
 cd /usr/local/cuda-8.0/samples/1_Utilities/deviceQuery  
 sudo make  
 ./deviceQuery  
      
 #編譯並測試頻寬 bandwidthTest：  
 cd ../bandwidthTest  
 sudo make  
 ./bandwidthTest  

如果這兩個測試的最後結果都是Result = PASS，說明CUDA安裝成功啦。

cudnn安裝

下載地址

具體方法參考我之前的部落格，注意修改一下啊，參考一下解壓cudnn後的資料夾，cuda/lib64/下的檔案。這就是不同點。

tensorflow-gpu安裝

首先得解除安裝之前的tensorflow

#啟用tensorflow環境
source activate tensorflow
pip uninstall tensorflow-gpu

pip install tensorflow-gpu