ubuntu安裝tensorflow gpu版本
阿新 • • 發佈:2018-11-19
由於之前用的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}}
成功
有個檔案是不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一下就好