jeston nano 安装pytorch

最新推荐文章于 2021-11-20 19:43:09 发布

chencaw

最新推荐文章于 2021-11-20 19:43:09 发布

阅读量1.3k

点赞数 4

CC 4.0 BY-SA版权

分类专栏： jeston nano

本文链接：https://blog.youkuaiyun.com/chencaw/article/details/117902535

jeston nano 专栏收录该内容

15 篇文章

订阅专栏

该博客指导如何在Jetson设备上安装PyTorch 1.8.0和torchvision 0.9.0。首先，通过官方或第三方链接下载适用于JetPack 4.4.x或4.5.x的Python 3.6的whl文件。然后，使用sudo apt-get安装必要的依赖，如pip和numpy。接着，使用pip3安装PyTorch whl文件。为安装torchvision，需要手动下载源码并使用python3 setup.py install进行安装。最后，通过Python脚本验证安装是否成功。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

（1）主要参考

http://blog.openpilot.cc/archives/2271

（2）官网下载地址

https://forums.developer.nvidia.com/t/pytorch-for-jetson-version-1-8-0-now-available/72048

（3）注意自己的镜像对应的版本，这里推荐下载1.8版本

PyTorch v1.8.0

JetPack 4.4 (L4T R32.4.3) / JetPack 4.4.1 (L4T R32.4.4) / JetPack 4.5 (L4T R32.5.0) / JetPack 4.5.1 (L4T R32.5.1)
- Python 3.6 - torch-1.8.0-cp36-cp36m-linux_aarch64.whl 3.4k

如果无法下载，进入如下好人的链接

https://download.youkuaiyun.com/download/qq_51095731/16753369

（3）安装方法

sudo apt-get install python3-pip libopenblas-base libopenmpi-dev 
pip3 install Cython
pip3 install numpy 
pip3 install pip3 install torch-1.8.0-cp36-cp36m-linux_aarch64.whl

(4)ceshixia

python3
import torch
print(torch.cuda.is_available())

（5）安装 torchvision

说明地址：https://github.com/pytorch/vision

因为我安装的是1.8.0的pytorch，对应 torchvision 版本是0.9.0，所以安装如下：、

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

#shangmian de  xian install
pip3 install torchvision==0.9.0  #我用这条没有安装成功

because pip3 install torchvision==0.9.0 #我用这条没有安装成功, 手动去这个地址下载代码包：

https://github.com/pytorch/vision/tree/v0.9.0，手动选择下载

解压后，进入文件夹

export BUILD_VERSION=0.9.0
sudo python3 setup.py install

一定要耐心等待，如果等不了，可以通过jtop看看情况

jtop

中间用到pillow>=4.1.1如果比较慢建议先安装

pip3 install -i https://pypi.doubanio.com/simple/ --trusted-host pypi.doubanio.com pillow

#测试是否安装成功
python3
import torchvision
print(torchvision.__version__)