【jetson笔记】torchaudio报错_导入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
1

运行docker镜像

sudo docker run -it --rm --runtime nvidia --network host nvcr.io/nvidia/l4t-ml:r35.1.0-py3
1

此时输入pip list就可以看到环境已安装好
ctrl+D可以退出docker命令行

如果需要挂载文件到docker镜像运行如下命令，并且在关闭后仍然在后台运行

sudo docker run -itd --runtime nvidia --network host -v /home/user/project:/location/in/container --name nvidia nvcr
.io/nvidia/l4t-ml:r35.1.0-py3
1
2

这样就可以把/home/user/project的文件挂载到docker的/location/in/container目录下
ctrl+D退出docker命令行后再次进入

docker exec -it nvidia /bin/bash
1

验证torch和torchaudio是否安装