jetson-xavier-nx安装pytorch

最新推荐文章于 2024-07-23 11:15:57 发布

BokyLiu

最新推荐文章于 2024-07-23 11:15:57 发布

阅读量1.2k

点赞数

分类专栏：环境搭建

原文链接：https://forums.developer.nvidia.com/t/pytorch-for-jetpack-4-4-l4t-r32-4-3-in-jetson-xavier-nx/141455

版权

环境搭建专栏收录该内容

5 篇文章 0 订阅

订阅专栏

安装PyTorch 1.6.0
wget https://nvidia.box.com/shared/static/yr6sjswn25z7oankw8zy1roow9cy5ur1.whl -O torch-1.6.0rc2-cp36-cp36m-linux_aarch64.whl
sudo apt-get install python3-pip libopenblas-base libopenmpi-dev
sudo pip3 install Cython
sudo pip3 install torch-1.6.0rc2-cp36-cp36m-linux_aarch64.whl

安装Torchvision 0.7.0
sudo apt-get install libjpeg-dev zlib1g-dev
git clone --branch v0.7.0 https://github.com/pytorch/vision torchvision
cd torchvision
sudo python3 setup.py install