Jetson tx2(JetPack 4.4)配置pytorch环境

2023-05-16

下载pytorch

  1. 下载pytorch1.7:我的系统是JetPack4.4,要求pytorch>=1.7.
    在这里插入图片描述

安装pytorch

sudo apt-get install libopenblas-base libopenmpi-dev
pip3 install torch-1.7.0-cp36-cp36m-linux_aarch64.whl

安装torchvision

  1. 这里选择torchvision 0.8.1(可根据自己的pytorch选择torchvision版本)
sudo apt-get install libjpeg-dev zlib1g-dev
git clone --branch v0.8.1 https://github.com/pytorch/vision torchvision   # see below for version of torchvision to download
cd torchvision
sudo python setup.py install
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系:hwhale#tublm.com(使用前将#替换为@)

Jetson tx2(JetPack 4.4)配置pytorch环境 的相关文章

随机推荐