Tensorflow

Tensorflow的安裝是以官方網站介紹為主

安裝虛擬環境

建議透過virtual environment方式安裝

安裝venv環境
$virtualenv --system-site-packages -p python3 ./venv

使用虛擬環境的方式是透過切換到另一個shell來進行

使用venv環境
$source ./venv/bin/activate

當你進入虛擬環境後,命令列提示字元將會出現(venv)

在這個虛擬環境下將可以任意安裝各種軟件包,而不會影響系統環境。

更新pip安裝
(venv)$pip install --upgrade pip
(venv)$pip list
離開虛擬環境
(venv)$deactivate

安裝Tensorflow

首先確認你要安裝的Tensorflow與CUDA版本

參考資訊

目前已安裝有CUDA9.0+cuDNN與CUDA10.0+cuDNN

請利用環境模組選擇合適的CUDA後再進行Tensorflow安裝

進入虛擬環境後開始進行Tensorflow安裝

安裝最新版tensorflow
(venv)$pip install install tensorflow-gpu
指定特地版本tensorflow
(venv)$pip install tensorflow-gpu==1.2.1

如果執行tensorflow時有出現錯誤訊息 可能是numpy版本問題 可透過pip重新安裝

重新安裝numpy
(venv)$pip uninstall numpy

(venv)$pip install numpy==1.16.4

參考為何GPU使用率上不去?