Jetson nx 上安装 torch torchvision报错fatal error: libavcodec/avcodec.h: No such file or directory

最新推荐文章于 2024-08-26 11:44:21 发布

鲤鱼不懂

最新推荐文章于 2024-08-26 11:44:21 发布

阅读量3.1k

点赞数

分类专栏： tensorrt jetson nx 文章标签： pytorch 人工智能 python

本文链接：https://blog.csdn.net/qq_43318374/article/details/121217034

版权

4 篇文章 0 订阅

订阅专栏

1 篇文章 0 订阅

订阅专栏

本文档记录了在NX设备上安装PyTorch和TorchVision的详细步骤，包括下载torch安装包，使用pip3安装，以及处理安装过程中遇到的libavcodec/avcodec.h文件缺失错误。解决方案是通过修改setup.py文件，屏蔽ffmpeg相关部分，成功完成安装。

摘要由CSDN通过智能技术生成

在nx上安装torch，先下载torch安装包。

wget https://nvidia.box.com/shared/static/9eptse6jyly1ggt9axbja2yrmj6pbarc.whl -O torch-1.6.0-cp36-cp36m-linux_aarch64.whl

安装torch命令：

sudo pip3 install Cython #防止安装torch报错
pip3 install torch-1.6.0-cp36-cp36m-linux_aarch64.whl

安装torchvision:
先安装依赖：

sudo apt-get install　libjpeg-dev zlib1g-dev　libopenblas-dev

下载安装包：

git clone --branch v0.7.0 https://github.com/pytorch/vision torchvision

安装步骤：

cd torchvision
export BUILD_VERSION=0.7.0
sudo python3 setup.py install

这个时候报错：

fatal error: libavcodec/avcodec.h: No such file or directory
 #include <libavcodec/avcodec.h>

在这里插入图片描述
我只需要把ffmpeg屏蔽掉就可以;修改setup.py

把 if has ffmpeg 修改为 if False

在这里插入图片描述

关注

专栏目录