我和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__)
相关推荐
寻星探路13 分钟前
测试开发话题10---自动化测试常用函数(2)
java·前端·python
大任视点18 分钟前
可梦AI获首批企业好评,蜜糖网络入驻共启AI短剧工业化
人工智能
鸢尾掠地平19 分钟前
Python中常用内置函数上【含代码理解】
开发语言·python
萧鼎24 分钟前
Python 图像处理利器:Pillow 深度详解与实战应用
图像处理·python·pillow
高洁0125 分钟前
大模型-详解 Vision Transformer (ViT)
人工智能·python·深度学习·算法·transformer
科技峰行者27 分钟前
亚马逊云科技与OpenAI战略合作深度分析:算力联盟重塑AI产业格局
人工智能
说私域31 分钟前
O2O行业风口下的运营策略与定制开发AI智能名片S2B2C商城小程序的应用研究
人工智能·小程序
慕慕涵雪月光白31 分钟前
在Ubuntu系统上安装英伟达(NVIDIA)RTX 3070 Ti的驱动程序
linux·运维·人工智能·ubuntu
柳鲲鹏34 分钟前
OpenCV:BGR/RGB转I420(颜色失真),再转NV12
人工智能·opencv·计算机视觉
无风听海36 分钟前
神经网络之线性变换
人工智能·深度学习·神经网络