码迷,mamicode.com
首页 > 其他好文 > 详细

xavier NX安装torchvision

时间:2020-07-10 15:38:19      阅读:186      评论:0      收藏:0      [点我收藏+]

标签:nload   eve   bsp   png   result   直接   viso   href   没有   

上一篇文章里面记录了在xavier NX上安装pytorch,https://www.cnblogs.com/cumtchw/p/13273753.html

然后接下来安装torchvision,下面是英伟达官网上的torchvision的安装方法:https://forums.developer.nvidia.com/t/pytorch-for-jetson-nano-version-1-5-0-now-available/72048

$ sudo apt-get install libjpeg-dev zlib1g-dev
$ git clone --branch <version> https://github.com/pytorch/vision torchvision   # see below for version of torchvision to download
$ cd torchvision
$ sudo python setup.py install
$ cd ../  # attempting to load torchvision from build dir will result in import error
$ pip install pillow<7 # always needed for Python 2.7, not needed torchvision v0.5.0+ with Python 3.6

然后下面是torch和torchvision的版本对应关系。

技术图片

 而我们安装的torch版本是1.6版本,并没有对应的torchvison,于是我直接从github上面下载最新的torchvision,https://github.com/pytorch/vision

然后用下面命令安装

$ cd torchvision
$ sudo python setup.py install
$ cd ../  # attempting to load torchvision from build dir will result in import error

 然后用下面的命令进行测试

python3
import torchvison
torchvision.__version__

xavier NX安装torchvision

标签:nload   eve   bsp   png   result   直接   viso   href   没有   

原文地址:https://www.cnblogs.com/cumtchw/p/13279051.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!