我和jetson-Nano的故事(12)——安装pytorch 以及 torchvision

在jetson nano中安装Anaconda、pytorch 以及 torchvision

1.Pytorch下载安装

  1. 首先登录英伟达官网下载Pytorch安装包,这里以PyTorch v1.10.0为例
  1. 安装依赖库
c 复制代码
sudo apt-get install libjpeg-dev zlib1g-dev libpython3-dev libavcodec-dev libavformat-dev libswscale-dev libopenblas-base libopenmpi-dev
  1. 安装Pytorch
c 复制代码
sudo chmod a+x torch-1.10.0-cp36-cp36m-linux_aarch64.whl
sudo pip3  install torch-1.10.0-cp36-cp36m-linux_aarch64.whl 

安装成功(注意Python版本得是3.6的)

  1. 验证Pytorch是否安装成功
c 复制代码
python3
import torch
x = torch.rand(5, 3)
print(x)
  1. 查看版本信息
c 复制代码
import torch
print(torch.__version__)

2.Torchvision安装

Torchvision版本要与Pytorch版本相匹配,我们前面安装的Pytorch版本为1.10.0Torchvision安装 v0.10.0版本

  1. Torchvision版本要与Pytorch版本相匹配,我们前面安装的Pytorch版本为1.10.0,Torchvision安装 v0.11.0版本
  2. 下载、安装torchvision
c 复制代码
git clone --branch v0.11.0 https://github.com/pytorch/vision torchvision
cd torchvision
export BUILD_VERSION=0.11.0
sudo python3 setup.py install

出现报错可能是Pillow版本太高,卸载重装

cpp 复制代码
sudo pip3 uninstall pillow
sudo pip3 install pillow
  1. 查看版本信息
cpp 复制代码
import torchvision
print(torchvision.__version__)
相关推荐
阿坡RPA9 小时前
手搓MCP客户端&服务端:从零到实战极速了解MCP是什么?
人工智能·aigc
用户27784491049939 小时前
借助DeepSeek智能生成测试用例:从提示词到Excel表格的全流程实践
人工智能·python
机器之心9 小时前
刚刚,DeepSeek公布推理时Scaling新论文,R2要来了?
人工智能
算AI11 小时前
人工智能+牙科:临床应用中的几个问题
人工智能·算法
JavaEdge在掘金11 小时前
ssl.SSLCertVerificationError报错解决方案
python
我不会编程55512 小时前
Python Cookbook-5.1 对字典排序
开发语言·数据结构·python
凯子坚持 c12 小时前
基于飞桨框架3.0本地DeepSeek-R1蒸馏版部署实战
人工智能·paddlepaddle
老歌老听老掉牙12 小时前
平面旋转与交线投影夹角计算
python·线性代数·平面·sympy
满怀101512 小时前
Python入门(7):模块
python
无名之逆12 小时前
Rust 开发提效神器:lombok-macros 宏库
服务器·开发语言·前端·数据库·后端·python·rust