ubuntu配置gpu版tensorflow

1.ubuntu下載下傳安裝anaconda

wget https://repo.continuum.io/archive/Anaconda3-2019.10-Linux-x86_64.sh
           

我下載下傳的是python3.7 版本對應的anaconda，不知道該安裝什麼版本可以找找anaconda和python的版本對應表

下載下傳好後，進行安裝

bash Anaconda3-2019.10-Linux-x86_64.sh
           

2.配置環境變量

（1）配置環境變量 輸入vim ~/.bashrc，按向下箭頭将光标調到最後一行，然後輸入 i 進入編輯模式，輸入指令export PATH="/home/ubuntu/anaconda3/bin:$PATH",再按esc,輸入 :wq 儲存退出。 下一次重新登入伺服器時輸入 source ~/.bashrc即可

（2）輸入 conda --version 檢查是否安裝成功

3. 建立虛拟環境

建立虛拟環境，命名為tf，python版本為3.7

conda create -n tf python=3.7
           

4. 激活虛拟環境

conda activate tf
           

5. 下載下傳cuda和cudnn

tensorflow對應的cuda和cudnn版本很重要

這裡使用1.14.0的tensorflow，是以選擇10.0的cuda和7.6.4的cudnn

（1）下載下傳cuda：

（2）下載下傳cudnn

有的版本會自動安裝cudnn，如果沒有自動安裝就執行以下指令安裝

6. gpu版tensorflow

（1）下載下傳tensorflow

pip install -i https://pypi.douban.com/simple/ --upgrade tensorflow==1.14.0
           

（2）下載下傳tensorflow-gpu

pip install -i https://pypi.douban.com/simple/ --upgrade tensorflow-gpu==1.14.0
           

參考部落格：

https://blog.csdn.net/jiuliang1916/article/details/79225711

https://blog.csdn.net/qq_43477550/article/details/103446745

https://blog.csdn.net/weixin_40588315/article/details/85881338