Jetson AGX Xavier配置PyTorch与TorchVision流程

1 环境

Jetson AGX Xavier，JetPack 4.4 production release (L4T R32.4.3)，Python 3.6.9。

需要安装的PyTorch与TorchVision对应的版本号为：

PyTorch v1.6 + torchvision v0.7.0-rc2

PyTorch与JetPack匹配的版本号：

PyTorch与TorchVision匹配的版本号：

2 安装PyTorch

PyTorch v1.6的wheels下载链接为：

https://nvidia.box.com/shared/static/yr6sjswn25z7oankw8zy1roow9cy5ur1.whl

安装命令为：

sudo apt-get install python3-pip libopenblas-base libopenmpi-dev

pip3 install Cython

pip3 install numpy ~/下载/torch-1.6.0rc2-cp36-cp36m-linux_aarch64.whl

3 安装TorchVision

安装命令为：

sudo apt-get install libjpeg-dev zlib1g-dev

git clone --branch v0.7.0-rc2 https://github.com/pytorch/vision torchvision

cd torchvision

sudo python3 setup.py install

cd ../

4 测试PyTorch与TorchVision

输入下列命令：

python3

import torch

print(torch.__version__)

print('CUDA available: ' + str(torch.cuda.is_available()))

print('cuDNN version: ' + str(torch.backends.cudnn.version()))

a = torch.cuda.FloatTensor(2).zero_()

print('Tensor a = ' + str(a))

b = torch.randn(2).cuda()

print('Tensor b = ' + str(b))

c = a + b

print('Tensor c = ' + str(c))

import torchvision

print(torchvision.__version__)

安装成功后的测试结果如下图所示：

5 参考文献

1. https://forums.developer.nvidia.com/t/pytorch-for-jetson-nano-version-1-5-0-now-available/72048
2. 查看jetpack的版本：https://www.jianshu.com/p/7f2d8a563ded
3. Linux下查看python版本号：https://blog.csdn.net/huangzx3/article/details/80067417

Jetson AGX Xavier配置PyTorch与TorchVision流程相关推荐

1. NVIDIA Jetson AGX Xavier 配置pytorch环境,anaconda,Pycharm

   提示:文章写完后,目录可以自动生成,如何生成可参考右边的帮助文档 概览 1.前言 2.配置版本 3.主环境的配置 4.配置过程中的坑,及对应解决方法 4.1.问题1 (1)原因分析 (2)解决方案 4 ...

2. Jetson AGX Xavier配置cuDNN流程

   Jetson AGX Xavier配置cuDNN流程 环境:Jetson AGX Xavier,CUDA 10.2,cuDNN 8.0.0 1 软件版本查询 CUDA版本查询: nvcc -V 或者, ...

3. Jetson Agx Xavier 安装cuda10.2和cudnn,pytorch,pycharm, unet

   1.refer to Jetson Agx Xavier 安装cuda10.2和cudnn_tianzhenba的博客-CSDN博客_agx 安装cuda bash cuda-l4t.sh cuda- ...

4. Jetson AGX Xavier刷机及环境配置

   写在前面:近期打算做一下视觉算法部署的内容,正好实验室有几个Xavier控制器,于是拿来用了一下,从零接触开始nvidia的硬件,以此记录学习过程. 一.Jetson AGX Xavier刷机 网上教 ...

5. jetson agx xavier：从亮机到yolov5下tensorrt加速

   重要的下载资源链接放在前面: jetpack4.5资源 主要内容记录在了自己的石墨文档里,自己习惯性地修改起来比较快,可能后续小修小改在那边更新.这里就做一个csdn的拷贝造福各位. https:// ...

6. Jetson AGX Xavier实现TensorRT加速YOLOv5进行实时检测

   上一篇:Jetson AGX Xavier安装torch.torchvision且成功运行yolov5算法 下一篇:Jetson AGX Xavier测试YOLOv4 一.前言 由于YOLOv5在Xa ...

7. Jetson Xavier NX and Jetson AGX Xavier Series启动流程

   文章目录 BootROM BootLoader Components加载程序组成 通用驱动程序框架 Microboot1, MB1 TegraBoot TegraBoot BPMP TegraBoot ...

8. 解决英伟达NVIDIA Jetson AGX Xavier Desktop Sharing无法运行 + Xavier 配置远程桌面

   解决英伟达NVIDIA Jetson AGX Xavier Desktop Sharing无法运行 + Xavier 配置远程桌面 修复Desktop Sharing无法打开 编译生效 配置Deskt ...

9. Jetson AGX Xavier 开机风扇自启动配置

   Jetson AGX Xavier 开机风扇自启动配置 安装jtop 按照如下配置: 注意 Fan Mode要设置为Default jetson_clocks和boot都要打开 Fan Speed设置 ...

最新文章

1. hook情况下,解决内联没有:hover的方案
2. Home界面的ListView
3. wpf 切换搜狗输入法英文_CPA《财管》考试，没时间按计算器，用搜狗V模式，真香！...
4. mysql 环形主从_【每日一博】MySQL 互为主从(环形结构)_MySQL
5. R的数据可视化，各种图表，常用统计量计算
6. 09. Django基础：URL反向解析
7. phpStudy JspStudy 2016 更新下载，新版支持php7.0
8. Java制作一个更加真实的按钮
9. 创业的几条至理名言，每条对创业者都大有帮助，值得收藏！
10. 2011黑帽大会：由黑客操控的世界
11. ios开发的p12和provision
12. pom.springmvc.psring-contect自用常用文件配置
13. MATLAB分析频谱
14. LiveData-原理全解析
15. python调用PHP方法
16. 工业交换机与商用交换机的区别有哪些
17. revit二次开发2016
18. 解析几何----拉格朗日乘数法在解析几何中的应用
19. 93---Python 极坐标系下绘制椭圆
20. 出生日期输入输出------C语言

热门文章

1. div图片垂直居中 如何使div中图片垂直居中
2. 【转载】gcc 使用中常用的参数及命令
3. ToDictionary
4. 比那 施氏食狮史 还要强的文章
5. ie模式下html中的按钮失效,解决ewebeditor编辑器在IE8下按钮失效、点击后没反应的问题...
6. php80端口改成8080,jetty,_JETTY 8080端口改为80端口无法访问，jetty - phpStudy
7. php使用hset报错,redis中set和hset的区别和使用场景
8. java samphore 连接池_ADOConnection数据库连接池
9. 安川e7变频器接线_台达变频器C2000系列在铝箔纸压花机上的应用
10. android 百度转码,移动端禁止百度自动转码的方法