我和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__)
相关推荐
不知更鸟2 分钟前
前端报错:快速解决Django接口404问题
前端·python·django
4***72136 分钟前
【玩转全栈】----Django模板语法、请求与响应
数据库·python·django
梁正雄18 分钟前
1、python基础语法
开发语言·python
Chat_zhanggong34534 分钟前
K4A8G165WC-BITD产品推荐
人工智能·嵌入式硬件·算法
霍格沃兹软件测试开发36 分钟前
Playwright MCP浏览器自动化指南:让AI精准理解你的命令
运维·人工智能·自动化
强化学习与机器人控制仿真42 分钟前
RSL-RL:开源人形机器人强化学习控制研究库
开发语言·人工智能·stm32·神经网络·机器人·强化学习·模仿学习
网易智企1 小时前
智能玩具新纪元:一个AI能力底座开启创新“加速度”
人工智能·microsoft
ituff1 小时前
微软认证考试又免费了
后端·python·flask
咚咚王者1 小时前
人工智能之数据分析 numpy:第十二章 数据持久化
人工智能·数据分析·numpy
沛沛老爹1 小时前
AI应用入门之LangChain中SerpAPI、LLM-Math等Tools的集成方法实践
人工智能·langchain·llm·ai入门·serpapi