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
sudo pip install --extra-index-url=https://developer.download.nvidia.com/compute/redist/jp33 tensorflow-gpu

tensorflow-nvJetson

安装keras

安装keras之前,一定要先把hdf5装一下,因为装的过程中会找不到hdf5,那就

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

然后

sudo pip install keras

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

Prefrence

Jetson tx2的tensorflow keras环境搭建

你可能感兴趣的:(TX2)