1. 程式人生 > >ubuntu安裝tensorflow gpu版本

ubuntu安裝tensorflow gpu版本

由於之前用的1.4,cuda8.0,所以這次重灌了一下、

先裝驅動

再裝cuda9.0,下載

sudo sh cuda_9.2.148_396.37_linux.run

Do you accept the previously read EULA,  輸入 accept,進入下一步。
INSTALL NVIDIA Accelerates Graphics Driver for Linux-x86_64 375.26,  輸入n,進入下一步;
Install the  CUDA 9.0 Toolkit?,  輸入y, 進入下一步;
Enter Tookit Location 點選enter, 進入下一步;
Do you want to install a symbolic link at /usr/local/cuda,  輸入y,進入下一步;
Install the CUDA 9.0 Samples?  輸入y, 進入下一步;
Enter CUDA Samples Location 點選enter, 進入下一步;

內容

export CUDA_HOME=/usr/local/cuda
export PATH=$PATH:$CUDA_HOME/bin
export LD_LIBRARY_PATH=/usr/local/cuda-9.0/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

   

成功

下載cudann

有個檔案是不tgz格式

cp cudnn-8.0-linux-x64-v5.1.solitairetheme8 cudnn-8.0-linux-x64-v5.1.tgz

tar -xvf cudnn-8.0-linux-x64-v5.1.tgz

安裝照著官網來

最後照著官網來裝tensorflow pip一下就好