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