其實我一直都在想,搞算法的不僅僅是服務,我們更是要在一個平台上去實現服務,因此,在工業領域,板子是很重要的,它承載着無限的機遇和挑戰,當然,我並不是特別懂一些底層的東西,但是這篇博客希望可以幫助有需要的人。
首先我們回到原點,就是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等等,缺啥裝啥就行了。