我和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__)
相关推荐
white-persist15 分钟前
Python实例方法与Python类的构造方法全解析
开发语言·前端·python·原型模式
Java 码农40 分钟前
Centos7 maven 安装
java·python·centos·maven
格林威44 分钟前
常规线扫描镜头有哪些类型?能做什么?
人工智能·深度学习·数码相机·算法·计算机视觉·视觉检测·工业镜头
lyx33136967591 小时前
#深度学习基础:神经网络基础与PyTorch
pytorch·深度学习·神经网络·参数初始化
倔强青铜三1 小时前
苦练Python第63天:零基础玩转TOML配置读写,tomllib模块实战
人工智能·python·面试
递归不收敛2 小时前
吴恩达机器学习课程(PyTorch 适配)学习笔记:3.3 推荐系统全面解析
pytorch·学习·机器学习
浔川python社2 小时前
《网络爬虫技术规范与应用指南系列》(xc—3):合规实操与场景落地
python
B站计算机毕业设计之家2 小时前
智慧交通项目:Python+YOLOv8 实时交通标志系统 深度学习实战(TT100K+PySide6 源码+文档)✅
人工智能·python·深度学习·yolo·计算机视觉·智慧交通·交通标志
高工智能汽车2 小时前
棱镜观察|极氪销量遇阻?千里智驾左手服务吉利、右手对标华为
人工智能·华为
IT森林里的程序猿2 小时前
基于机器学习方法的网球比赛胜负趋势预测
python·机器学习·django