【jetson笔记】torchaudio报错

原因是因为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.2.1-py3

这样就可以把/home/user/project的文件挂载到docker的/location/in/container目录下

验证torch和torchaudio是否安装