jetson tx2 nano配置pytorch环境记录笔记

自从买回来开发板后,一直在搞这个pytorch环境,老是出问题,今天又尝试配置了一下,成功了,很激动,自此记录一下,方便自己看,也希望能给后面人一个思路吧。废话不多说,开始搞。另外,通过通过的帖子配成功的话,希望给个赞!!



在jetson tx2 nano开发板中由于一些环境是自动烧入的,所以cuda是10.2的,我看的别的帖子cuda10.2对应的torch和torchvision版本是1.8.0与0.9.0的,我就下载一顿安装,导致一直有问题,后来通过命令安装时发现现在torchvision是0.9.1,我试着下载还真是,成功了。下面看的可能不太清,但是显示的是没有0.9.0版本的。

 所以要准备的torch=1.8.0,torchvision=0.9.1的。torchvision文件可以从这个网址下载http://download.pytorch.org/whl/torchvision可以ctrl+f搜索“torchvision-0.9.1-cp36-cp36m-manylinux2014_aarch64.whl”。torch版本可以自己找一下,找不到的话可以私信我。

pip3 install



 在安装之前除了上面需要的文件准备好之外,需要先看一下你的pip版本,初始版本是9.1的,这里需要升级一下pip,注意了,注意了!!这里有个大坑,不要跟着命令行中提示的升级。你运行下面的命令,

python3 -m pip install --upgrade pip

安装完成后输入pip3 -V看看pip版本,应该为21.3.1版本的,比这个高的,恭喜你,你可以试着解决一下,解决不了的,你需要重新刷系统,你跟我的命令的话,一般不会有错。这里小说一下原因,因为python3.6.9支持的pip版本最高到21版本的。

当你升高pip后,会有一个警告,你可能会认为它是一个警告而小看它,如果你是这麽想的,你就又的多走弯路,不信我,可以试试,这都是血的教训。

这里我把解决办法整理成另一个博客了,下面是网址没法复制,你可以去我的博客翻翻。

https://mp.csdn.net/mp_blog/creation/editor/131068982应该是这个。



升级完pip后,开始安装torch,下面三个命令逐一运行,其中,torch-1.8.0-cp36-cp36m-linux_aarch64.whl为开始准备的文件。

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

pip3 install Cython

pip3 install numpy torch-1.8.0-cp36-cp36m-linux_aarch64.whl

安装完成之后可以检验一下,可以输入python3,之后输入看看版本是否对应。

>>> import torch
>>> torch.__version__



安装torchvision,下面的命令逐一运行

$ sudo apt-get install libjpeg-dev zlib1g-dev libpython3-dev
$ sudo apt-get install libavcodec-dev libavformat-dev libswscale-dev
$ sudo pip3 install -U pillow

$pip3 install torchvision-0.9.1-cp36-cp36m-manylinux2014_aarch64.whl

安装后

>>> import torchvision
>>> torchvision.__version__
看看版本号对应不。至此完成。。。。。。。。。。。。。。。

猜你喜欢

转载自blog.csdn.net/qq_43384629/article/details/131074812