Jetson tx2的tensorflow keras環境搭建


其實我一直都在想,搞算法的不僅僅是服務,我們更是要在一個平台上去實現服務,因此,在工業領域,板子是很重要的,它承載着無限的機遇和挑戰,當然,我並不是特別懂一些底層的東西,但是這篇博客希望可以幫助有需要的人。

首先我們回到原點,就是jetpack 3.3刷完機后,現在要裝tensorflow和keras。自然的,我們可以想到,需要

  • miniconda或anaconda
  • cuda和cudnn
  • tensorflow
  • keras

其實jetpack3.3里面已經有了python2和cuda9.0,cudnn7這些都是對應綁定的關系,如果你胡亂裝肯定是不能搞的,所以問題變得簡單了,就是裝tensorflow和keras,當然很多沒有的包就通過pip安裝,miniconda、anaconda就不要考慮了。

下面正式開始:

  • 安裝Java
sudo add-apt-repository ppa:webupd8team/java 
sudo apt-get update 
sudo apt-get install oracle-java8-installer

如果apt-get出現Unable to lock the administration directory的錯誤就

sudo rm /var/cache/apt/archives/lock
sudo rm /var/lib/dpkg/lock

如果還不行,關閉終端重新打開。

  • 安裝其它庫
sudo apt-get install zip unzip autoconf automake libtool curl zlib1g-dev maven -y
  • 安裝bazel

上這個網站 ,下載bazel 0.10.0,然后將其解壓成bazel-dist文件夾,開始編譯:

sudo chmod -R ug+rwx bazel-dist 
cd bazel-dist
./compile.sh  

編譯完成,執行:

sudo cp output/bazel /usr/local/bin
  • 創建虛擬內存
fallocate -l 8G swapfile
chmod 600 swapfile
mkswap swapfile
sudo swapon swapfile
swapon -s
  • 安裝tensorflow-gpu

打開項目,可以發現下圖:

意思就是裝tensorflow 1.10就好了,這個時候,我們用項目里面release好的版本,地址下載1.10的whl文件。然后

sudo pip install 文件名

不出意外,就裝好了。可以去我之前的博客找個例子試一試。

  • 安裝keras

安裝keras之前,一定要先把hdf5裝一下,因為裝的過程中會找不到hdf5,那就

sudo apt-get install libhdf5-dev
sudo apt-get install python-h5py

然后

sudo pip install keras

不出意外,就搞定了。除此之外我們可能需要更多的python模塊,比如scipy,matplotlib,sklearn等等,缺啥裝啥就行了。


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM