Jetson 共用资料

Torch&&Torchvision安装

2023-07-21 admin 170

Torch&&Torchvision安装

通过jtop查询的系统配置如图所示:

 

 

1.安装torch

根据下面的百度网盘下载对应版本的离线安装包

链接:https://pan.baidu.com/s/1zpqH8GUJ06UlxoUd6hMzpQ

提取码:ldbp

下载完成后使用U盘复制到设备上,执行以下命令安装(以下以)

sudo apt-get install python3-pip libopenblas-base libopenmpi-dev libomp-dev

pip3 install Cython

pip3 install numpy torch-1.12.0-cp36-cp36m-linux_aarch64.whl

 

2.安装torchvision

通过下图找到与torch相对应的版本,发现torch是1.12,那么torchvision即为0.13.0

 

输入以下命令

sudo apt-get install libjpeg-dev zlib1g-dev libpython3-dev libavcodec-dev libavformat-dev libswscale-dev

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

cd torchvision

export BUILD_VERSION=0.13.0

python3 setup.py install --user

等待安装完成即可

 

 

附录

其它参考链接:

https://forums.developer.nvidia.com/t/pytorch-for-jetson/72048


首页
产品
案例
联系