K_程式人

本版安裝Tensorflow GPU版本為官方提供的,

假設TX2是已經刷完機以及安裝完cuda等套件,

可以發現已經內建兩種版本的Python,

分別是Python2.7及Python3.5,

再安裝Tensorflow gpu前,

要先安裝pip套件,

首先先更新系統

sudo apt-get update

sudo apt-get upgrade

因本版是使用Python3.5, 因此安裝pip時要安裝python3的版本

sudo apt install python3-pip 

安裝完的pip版本為8.1.1

未來要使用pip套件要輸入 pip3 install xxx

※如果pip套件是用最新版本, 執行出現error請參考此篇↓※

修正pip套件執行Error問題 ←可直接點擊

接著安裝官方的tensorflow-gpu的套件,

sudo apt-get update

pip3 install --extra-index-url https://developer.download.nvidia.com/compute/redist/jp33 tensorflow-gpu

安裝完成後, 進入python編譯環境

python3

import tensorflow

如輸入完沒出現任何error表示已成功安裝,

移除Tensorflow

pip3 uninstall -y tensorflow-gpu