如何在JETSON上运行tensorflow 与 pytorch？

pytorch: https://github.com/pytorch/pytorch#nvidia-jetson-platforms

tensorflow: https://docs.nvidia.com/deeplearning/frameworks/install-tf-jetson-platform/index.html#install

您好，我们按照 https://docs.nvidia.com/deeplearning/frameworks/install-tf-jetson-platform/index.html#install安装tensorflow，报错如截图

方凯 发表于 2020-4-27 11:53
您好，我们按照 https://docs.nvidia.com/deeplearning/frameworks/install-tf-jetson-platform/index.html ...

这个问题tensorflow serving的docker在jetson平台不兼容导致的。您可以尝试在这个连接里面找一下您需要的docker尝试一下：https://hub.docker.com/u/helmuthva

Geary 发表于 2020-4-27 19:54
这个问题tensorflow serving的docker在jetson平台不兼容导致的。您可以尝试在这个连接里面找一下您需要的d ...

您好，我按照链接下载了Docker，安装过程中有报错，能否再帮忙看一下。

方凯 发表于 2020-5-6 11:01
您好，我按照链接下载了Docker，安装过程中有报错，能否再帮忙看一下。

您好：

如果是安装pytorch和tensorflow，需要注意pytorch和tensorflow的版本，和Jetpack的版本匹配。
您可以在如下链接找到对应的tensorflow和pytorch的版本，然后下载来进行安装。
https://elinux.org/Jetson_Zoo#TensorFlow

这种方法是由Nvidia官方来确保的，所以最推荐。

目前您参考的
https://docs.nvidia.com/deeplear ... /index.html#install，上面是针对JP4.4版本的命令，而JP4.4把cuda升级到了10.2，所以会出现不兼容的问题。但是JP4.4目前还是DP版本（开发者预览版本），坑比较多，如果不是为了尝鲜我们建议可以等一等。等到6月初JP4.4正式版本放出以后再进行尝试。

关于https://hub.docker.com/u/helmuthva的docker
由于不是NV提供的，我看到目前支持的JP版本是4.2.1，而您使用镜像应该是4.2.2的。我不太确定是否是因为这个原因造成的问题，等我有空的时候会尝试一下。然后反馈给您。

方凯 发表于 2020-5-6 11:01
您好，我按照链接下载了Docker，安装过程中有报错，能否再帮忙看一下。

不知道您这边启动docker的时候是不是使用了 nvidia-docker 来替代docker命令，NV提供的nvidia-docker对docker做了一些处理，将GPU资源映射到docker内了，如果直接使用docker应该是有些资源在docker中无法直接访问的。你可以试试nvidia-docker命令启动docker。

这个命令可以很方便的在JP4.2上安装tensorflow，应该比docker要方便一些。

sudo pip3 install --pre --extra-index-url https://developer.download.nvidia.com/compute/redist/jp/v42 tensorflow-gpu==1.14.0+nv19.10