Jetson orin 安装tensorflow和pytorch

1.安装tensorflow, 安装官方方法，基本一步一步能够完成。

Installing TensorFlow for Jetson Platform :: NVIDIA Deep Learning Frameworks Documentation

2.安装pytorch，安装官方方法，中间不少问题

Installing PyTorch for Jetson Platform :: NVIDIA Deep Learning Frameworks Documentation

第一:

Export with the following command:

export TORCH_INSTALL=https://developer.download.nvidia.cn/compute/redist/jp/v502/pytorch/1.13.0a0+936e930.nv22.11

这个会找不到这个安装包，直接打开网站：https://developer.download.nvidia.cn/compute/redist/jp/v502/pytorch/，然后选择一个安装包的目录，我使用的是：torch-1.13.0a0+936e9305.nv22.11-cp38-cp38-linux_aarch64.whl

第二：

torchvision使用pip3 install torchvision可能会存在版本和torch不兼容问题：

可去官网查看：github.com/pytorch/vision#installation中查看

但选择正确版本后，仍然会提示，UserWaring: Failed to load image Python extension: {e},    网上找了不少方法，仍无法解决，但因为只是个Warning，可以不用管。

第三

跑yolov7时，仍会出错，提示可能torchvision版本和torch不兼容。

卸载torchvision,使用源码安装，原因可能pip3安装的不是linux_aarch64版本。

$ export BUILD_VERSION=0.14.0  # where 0.x.0 is the torchvision version  
$ python setup.py install user     # use python3 if installing for Python 3.6

如果担心pillow问题，可以重新安装pillow:

 pip3 install --upgrade pillow -i https://pypi.tuna.tsinghua.edu.cn/simple