Jetson tx2的tensorflow keras环境搭建

2021年11月21日 阅读数:5
这篇文章主要向大家介绍Jetson tx2的tensorflow keras环境搭建,主要内容包括基础应用、实用技巧、原理机制等方面,希望对大家有所帮助。

其实我一直都在想,搞算法的不单单是服务,咱们更是要在一个平台上去实现服务,所以,在工业领域,板子是很重要的,它承载着无限的机遇和挑战,固然,我并非特别懂一些底层的东西,可是这篇博客但愿能够帮助有须要的人。java

首先咱们回到原点,就是jetpack 3.3刷完机后,如今要装tensorflow和keras。天然的,咱们能够想到,须要python

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

其实jetpack3.3里面已经有了python2和cuda9.0,cudnn7这些都是对应绑定的关系,若是你胡乱装确定是不能搞的,因此问题变得简单了,就是装tensorflow和keras,固然不少没有的包就经过pip安装,miniconda、anaconda就不要考虑了。git

下面正式开始:github

  • 安装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的错误就web

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文件夹,开始编译:oracle

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

编译完成,执行:curl

sudo cp output/bazel /usr/local/bin
  • 建立虚拟内存
fallocate -l 8G swapfile
chmod 600 swapfile
mkswap swapfile
sudo swapon swapfile
swapon -s
  • 安装tensorflow-gpu

打开项目,能够发现下图:maven

意思就是装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等等,缺啥装啥就好了。