原因是因为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)
bash
sudo docker pull nvcr.io/nvidia/l4t-ml:r35.1.0-py3
运行docker镜像
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
目录下