Jetson TX2 安装 Tensorflow

版权声明:本文为博主原创文章,转载需标注来源。 https://blog.csdn.net/Code_Mart/article/details/82260271

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

方法一:安装 NVIDIA 官方发布文件

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

#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 的开发者网站逛逛,上面有许多问题的解决办法。

方法二:安装他人已编译好的文件

在 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

安装他人所编译好的文件

方法三:源码编译

1.设置虚拟内存

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

执行第五步,输出中会有新建立的虚拟空间,否则确认是否正确执行了上面的命令。

备注:不设置虚拟内存会出现在TensorFlow编译时,内存不足导致的编译失败

2.安装Java

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

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

4. 下载 Tensorflow 源码并编译

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

参考资料:
Jetson TX2 安装TensorFlow深度学习框架
在 Nvidia Jetson TX2 上编译安装tensorflow
在NVIDIA Jetson TX2上安装TensorFlow
Jetson TX2刷机及安装tensorflow gpu注意事项

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

作者信息:
CSDN:Code_Mart
Github:Bojack-want-drink

猜你喜欢

转载自blog.csdn.net/Code_Mart/article/details/82260271