原因是因为pip安装的包与jetson不兼容导致
自己安装或者cmake编译也会报错
需要拉取官方配置好的docker镜像
拉取docker镜像
具体容器可以看官网，按照自己需求拉取即可
https://catalog.ngc.nvidia.com/orgs/nvidia/containers/l4t-ml
如果其他包不需要只需要torch和torchaudio就拉取下面
https://catalog.ngc.nvidia.com/orgs/nvidia/containers/l4t-pytorch
我这边是JetPack 5.0.2 (L4T R35.1.0)
sudo docker pull nvcr.io/nvidia/l4t-ml:r35.1.0-py3
运行docker镜像
sudo docker run -it --rm --runtime nvidia --network host nvcr.io/nvidia/l4t-ml:r35.1.0-py3
此时输入pip list就可以看到环境已安装好
ctrl+D可以退出docker命令行
如果需要挂载文件到docker镜像运行如下命令
sudo docker run -it --rm --runtime nvidia --network host -v /home/user/project:/location/in/container nvcr.io/nvidia/l4t-ml:r35.1.0-py3
这样就可以把/home/user/project的文件挂载到docker的/location/in/container目录下
验证torch和torchaudio是否安装
