[日常] Jetson TX2 安装 Tensorflow

在Jetson TX2 上安装 Tensorflow 是一件令人烦躁的事情,但是最近 NVIDIA 终于为大家提供了官方版本,此外,我们还可以通过编译源码或者安装他人已编译好的文件进行安装。

目录

  1. 安装 NVIDIA 官方发布文件
  2. 安装他人已编译好的文件
  3. 源码编译

1. 安装 NVIDIA 官方发布文件

最近,NVIDIA 官方发表文章,宣称为 Jetson TX2 的开发者们提供了 Tensorflow 官方版!(文章链接)

[日常] Jetson TX2 安装 Tensorflow_第1张图片
#For Python 2.7
pip install –extra-index-url=https://developer.download.nvidia.com/compute/redist/jp33 tensorflow-gpu

#For Python 3.5
pip3 install –extra-index-url=https://developer.download.nvidia.com/compute/redist/jp33 tensorflow-gpu

此外,我建议大家多去 Jetson 的开发者网站逛逛,上面有许多问题的解决办法。

2. 安装他人已编译好的文件

在 GitHub 上有人为此提出了解决办法,我们只需要下载他所提供的文件进行安装即可(项目链接)。
这里需要指出,我用的是 Jetson TX2,刷的是 JetPack 3.3,安装了 CUDA 9.0 和 cuDNN v7.1.5,所以我下载的是Tensorflow 1.10。当你使用这个办法时,你得根据作者的提示选择相应版本的 Tensorflow 文件进行安装。

# 安装命令 Tensorflow 1.10
sudo pip install tensorflow-1.7.0-cp27-cp27mu-linux_aarch64.whl
[日常] Jetson TX2 安装 Tensorflow_第2张图片

3. 源码编译

3.1. 设置虚拟内存
  • 创建8G大小的swapfile
fallocate -l 8G swapfile
  • 更改swapfile的权限
chmod 600 swapfile
  • 创建swap区
mkswap swapfile
  • 激活swap区
sudo swapon swapfile
  • 确认swap区已在使用
swapon -s

执行第五步,输出中会有新建立的虚拟空间,否则确认是否正确执行了上面的命令。
备注:不设置虚拟内存会出现在TensorFlow编译时,内存不足导致的编译失败

3.2. 安装Java
sudo add-apt-repository ppa:webupd8team/java
sudo apt-get update
sudo apt-get install oracle-java8-installer
3.3. 安装Bazal
# 安装JDK8
sudo apt-get install openjdk-8-jdk\
# 安装和更新Bazel
sudo apt-get update && sudo apt-get install bazel
sudo apt-get upgrade bazel

3.4. 下载 Tensorflow 源码并编译

由于我并没有使用这种方法,所以我没法进行详细的描述,大家可以参考这篇博客:Jetson TX2 安装TensorFlow深度学习框架

参考资料:

  • Jetson TX2 安装TensorFlow深度学习框架
  • 在 Nvidia Jetson TX2 上编译安装tensorflow
  • 在NVIDIA Jetson TX2上安装TensorFlow
  • Jetson TX2刷机及安装tensorflow gpu注意事项
  • JetsonTX2上安装tensorflow的心酸史

如果你看到了这篇文章的最后,并且觉得有帮助的话,麻烦你花几秒钟时间点个赞,或者受累在评论中指出我的错误。谢谢!

作者信息:
知乎:没头脑
LeetCode:Tao Pu
CSDN:Code_Mart
Github:Bojack-want-drink

你可能感兴趣的:(环境配置)