【jetson笔记】torchaudio报错

yyhnet.cn2024-01-25 16:55

原因是因为pip安装的包与jetson不兼容导致

自己安装或者cmake编译也会报错

需要拉取官方配置好的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)

bash 复制代码

sudo docker pull nvcr.io/nvidia/l4t-ml:r35.1.0-py3

bash 复制代码

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

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

如果需要挂载文件到docker镜像运行如下命令

bash 复制代码

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目录下