Jetson Nano/TX1/2/NX 安装pytorch

YuAnn予安

于 2020-12-07 17:34:42 发布

阅读量666

点赞数

分类专栏： TX2 文章标签： pytorch 机器学习神经网络

本文链接：https://blog.csdn.net/weixin_48695448/article/details/110822797

版权

本文介绍了如何在Jetson Nano/TX1/2/NX上安装PyTorch，包括选择不同版本的PyTorch和torchvision，以及详细安装步骤，特别提示1.6.0版本安装更稳定。

摘要由CSDN通过智能技术生成

Jetson Nano/TX1/2/NX 安装pytorch

1.选择版本
2.安装torch
- Python 3.6
- Python 2.7
选择torchvision版本
安装torchvision
检验安装

1.选择版本

NVIDIA官网链接：PyTorch pip wheels
在这里插入图片描述
注意我在安装1.7.0版本时候出现错误，经过测试更换为1.6.0版本则安装正常。

2.安装torch

Python 3.6

wget https://nvidia.box.com/shared/static/wa34qwrwtk9njtyarwt5nvo6imenfy26.whl -O torch-1.7.0-cp36-cp36m-linux_aarch64.whl
sudo apt-get install python3-pip libopenblas-base libopenmpi-dev 
pip3 install Cython
pip3 install numpy torch-1.7.0-cp36-cp36m-linux_aarch64.whl

Python 2.7

wget https://nvidia.box.com/shared/static/1v2cc4ro6zvsbu0p8h6qcuaqco1qcsif.whl -O torch-1.4.0-cp27-cp27mu-linux_aarch64.whl
sudo apt-get install libopenblas-base libopenmpi-dev 
pip install future torch-1.4.0-cp27-cp27mu-linux_aarch64.whl