NX配置pytorch,torchvision环境
1. 系统版本情况
NX系统:Ubuntu18.04
python版本:3.6.9
JetPack:4.4
cuda版本:10.2
安装的Pytorch版本为1.6.0,torch vision版本为0.7.0.
2. 安装pytorch
推荐网上下载whl安装包安装,安装包下载地址在这里,因为NX是arm架构的,所以这里下载aarch64的安装包,如:Python 3.6 - torch-1.6.0-cp36-cp36m-linux_aarch64.whl下载完成后,执行
pip3 install XXX
原创
2021-07-28 19:41:25 ·
259 阅读 ·
0 评论