Jetson TX2零基础学习(二)——安装pip3、pytorch、torchvision

目录

一、安装pip3

二、安装pytorch

三、安装torchvision

系列文章

大家好,很开心又见面了,这次接着上篇博客,为大家详细介绍在Jetson TX2中搭环境的方法。

一、安装pip3

首先要知道Jetson TX2板卡的系统中自带python3.6,无需再装,那么为了使用pip命令,我们先进行pip3的安装,打开命令行终端(Ctrl + Alt + T),输入下面的代码

sudo apt-get install python3-pip python3-dev

安装完之后查一下系统都安装了什么,看一下pip版本,如果低的话需要升级

pip3 list   #查询
python3 -m pip install  --upgrade pip  #升级pip

那么现在就安装好了pip3,可以像windows一样使用指令安装所需要的库了,不过要注意写指令时不能用pip,而是要用pip3。

二、安装pytorch

安装pytorch需要先确定安装方式和版本,最好采用离线安装方式:在主机下载好安装包,用U盘拷到板卡中再安装,版本的话需要看你要移植的程序的需求,一般项目文件夹都会有一个名字为requirements.txt的文件,打开它可以看到程序需要的各种库和版本

Jetson TX2零基础学习(二)——安装pip3、pytorch、torchvision

这是我的,可以看到torch和torchvision的版本最低要求,分别是1.7.0和0.8.1,我们先装这两个,在装好pycharm之后就要把所有需要的库装好,库之间都或多或少有联系, 一定要先装torch和torchvision,那么先去官网下载离线安装包

https://forums.developer.nvidia.com/t/pytorch-for-jetson-version-1-10-now-available/72048

Jetson TX2零基础学习(二)——安装pip3、pytorch、torchvision

我选择的是torch1.8.0 ,下载最好科学上网,否则很难连接。如果你没办法下载的话,我把我下载好的资源分享给大家:

链接:https://pan.baidu.com/s/1JFHB5y8MYWl0l2oD3nePfA
提取码:pwx0

Jetson TX2零基础学习(二)——安装pip3、pytorch、torchvision

里面有这四个文件,大家下载好了放U盘里,拷贝到板卡中,之后三个文件都会用到。因为sdkmanager是烧录用的,上篇文章介绍过。

拷贝完成后,目光回到板卡,接下来依次输入下面的命令

sudo apt-get install python3-pip libopenblas-base libopenmpi-dev
pip3 install Cython
pip3 install numpy
pip3 install torch-1.8.0-cp36-cp36m-linux_aarch64.whl #这是离线安装包名字

问Y/N 就输入Y+回车

torch的安装很快就会完成,然后用pip3 list查一下有没有安装好

三、安装torchvision

torchvision的版本必须与torch对应,版本对照很好查到,我截取一部分放在下面

Jetson TX2零基础学习(二)——安装pip3、pytorch、torchvision

所以我们选择的torch1.8.0对应的是torchvision0.9.0 ,torchvision的下载有多种方式,可以去github下载,也可以去网站下载,我选择的是第一种,那我先说一下第二种:

进入网站:https://download.pytorch.org/whl/torch_stable.html

里面东西很多,需要别的也可以来下载,点击就直接开启下载了

Jetson TX2零基础学习(二)——安装pip3、pytorch、torchvision

看我画红线的地方,首先看 torchvision版本(0.9.0),再看咱们的 python版本(3.6对应cp36),最后看 系统(linux),所以选择我画的这个,点击即可下载,速度很快,是和torch一样的whl文件,那么安装方式也相同,输入同样类型的命令即可

安装torchvision比安装torch要费劲些,还要再安装几个库才能安装,否则中途报错,输入指令:

sudo apt-get install libjpeg-dev zlib1g-dev libpython3-dev libavcodec-dev libavformat-dev libswscale-dev

因为需要pillow,提前换源,否则下不动,输入命令

pip3 install -i https://pypi.doubanio.com/simple/ --trusted-host pypi.doubanio.com pillow

之后就可以安装torchvision了

pip3 install torchvision-0.9.0+cpu-cp36-cp36m-linux_x86_64.whl #安装命令

第一种方法稍微麻烦点,进入下面链接

https://github.com/pytorch/vision/tree/v0.9.0

想要什么版本直接在网址最后改就行,比如我想要0.8.0的那就进入下面链接

https://github.com/pytorch/vision/tree/v0.8.0

Jetson TX2零基础学习(二)——安装pip3、pytorch、torchvision

点绿色的Code,然后点我画圈的地方,不需要科学上网。

Jetson TX2零基础学习(二)——安装pip3、pytorch、torchvision

下好了是和我网盘里的一样的,将这个压缩包拷贝进板卡中,右键解压(不知道哪个英文是解压的话去网上搜一下),然后进入解压好的文件夹,右键空白处选择open in terminal,输入:

sudo python3 setup.py install

在执行这条命令前,要把上面说过的安装库和pillow换源完成才行。安装vision时间很长,耐心等待即可,安装好了之后用pip3 list查一下,确认torch和torchvision都成功安装了之后就可以进行下一阶段工作了—— 安装PyCharm和相关库,下篇文章我再来介绍~

如果你按照我上面讲的步骤来安装,是可以避免很多BUG的,所以如果遇到问题先看看是不是落下哪个步骤了,不行的话再复制Error在站内搜索一下。

如果感觉有帮助的话,可以点个 关注 吗~

您的支持会让我充满动力!

系列文章

Jetson TX2零基础学习(一)——连线、刷机:http://t.csdn.cn/LNv0k

Original: https://blog.csdn.net/qq_47422985/article/details/124457777
Author: 七海Shark
Title: Jetson TX2零基础学习(二)——安装pip3、pytorch、torchvision

原创文章受到原创版权保护。转载请注明出处:https://www.johngo689.com/708687/

转载文章受原作者版权保护。转载请注明原作者出处!

(1)

大家都在看

亲爱的 Coder【最近整理,可免费获取】👉 最新必读书单  | 👏 面试题下载  | 🌎 免费的AI知识星球