栏目分类:
子分类:
返回
名师互学网用户登录
快速导航关闭
当前搜索
当前分类
子分类
实用工具
热门搜索
名师互学网 > IT > 软件开发 > 后端开发 > Python

Jetson xvaier nx 安装torch1.6.0 torchvision0.7.0

Python 更新时间: 发布时间: IT归档 最新发布 模块sitemap 名妆网 法律咨询 聚返吧 英语巴士网 伯小乐 网商动力

Jetson xvaier nx 安装torch1.6.0 torchvision0.7.0

已经安装了python3 pip3 

ubuntu18.04

jetpack4.5

cuda10.2

。。。。

1、安装PyTorch1.6.0

链接: https://pan.baidu.com/s/1c-N5JrWT0UbfratLNmzSiQ

提取码: evnm 


百度网盘下载 torch-1.6.0-cp36-cp36m-linux_aarch64.whl文件,之后在.whl文件路径下运行:

sudo apt-get install python3-pip libopenblas-base libopenmpi-dev 
pip install Cython
pip install numpy torch-1.6.0-cp36-cp36m-linux_aarch64.whl

2、安装TorchVision0.7.0
下载TorchVision0.7.0版本文件夹:

sudo apt-get install libjpeg-dev zlib1g-dev
git clone --branch v0.7.0 https://github.com/pytorch/vision vision-0.7.0

在vision-0.7.0路径下运行:(注意下载后源码要全部赋予读写权限)

sudo python setup.py install
 

3.安装torch2rt

apt-get install libopenblas-base libopenmpi-dev

pip3 install packaging -i https://pypi.tuna.tsinghua.edu.cn/simple

git clone https://github.com/NVIDIA-AI-IOT/torch2trt
cd torch2trt  (注意下载后源码要全部赋予读写权限)
sudo  python3 setup.py install
转载请注明:文章转载自 www.mshxw.com
本文地址:https://www.mshxw.com/it/991095.html
我们一直用心在做
关于我们 文章归档 网站地图 联系我们

版权所有 (c)2021-2022 MSHXW.COM

ICP备案号:晋ICP备2021003244-6号