TX2安裝tensorflow-gpu(親測有效)

安裝好JetPack 4.3之後,再也不需要從源碼對tensorflow進行編譯安裝,可以通過以下兩步安裝:
Step1: 安裝h5py

sudo apt-get install python3-h5py

主要原因還是我的tx2上pip3安裝h5py時會失敗.

Step2:安裝tensorflow-gpu=1.15.0

sudo pip3 install --pre --extra-index-url https://developer.download.nvidia.com/compute/redist/jp/v43 tensorflow-gpu==1.15.0+nv19.12

Step3:驗證安裝

import tensorflow as tf
tf.__version__
#1.15.0
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章