我和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__)
相关推荐
ID_180079054731 分钟前
Python调用淘宝评论API:从入门到首次采集全流程
服务器·数据库·python
小猪咪piggy3 分钟前
【Python】(2) 执行顺序控制语句
开发语言·python
Σdoughty5 分钟前
python第三次作业
开发语言·前端·python
zhihuaba6 分钟前
构建一个基于命令行的待办事项应用
jvm·数据库·python
MediaTea7 分钟前
Python:内置类型也是类对象
开发语言·python
InfiSight智睿视界8 分钟前
当老字号遇上AI:阳坊涮肉的数字化运营转型之路
大数据·人工智能·连锁店智能巡检
张3蜂10 分钟前
OpenClaw 深度解析:从个人 AI 助理到开源智能体平台
人工智能·开源
程序员欣宸10 分钟前
LangChain4j实战之十六:RAG (检索增强生成),Naive RAG
java·人工智能·ai·langchain4j
Dingdangcat8611 分钟前
轮胎缺陷检测与分类系统基于solov2_r101_fpn_ms-3x_coco模型实现_fulltyre专项识别_1
人工智能·分类·数据挖掘
Ivanqhz11 分钟前
现代异构高性能计算(HPC)集群节点架构
开发语言·人工智能·后端·算法·架构·云计算·边缘计算