Torch&&Torchvision安装
2023-07-21
admin
173
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