Bootstrap

【玩转Jetson TX2 NX】(十)TX2 NX 安装Archiconda3+创建pytorch环境(详细教程+错误解决)

1 写在前面的话

该篇文章创建pytorch环境部分,大家谨慎参考。本以为已经完全解决错误,但实际上pytorch最终并未配置在所创建虚拟环境中,本人能力有限,实在搞不定,故大家可在此博客的踩坑基础上,继续前行,争取一次创建成功,加油!

2 TX2 NX 安装Archiconda3

Archiconda下载链接: https://github.com/Archiconda/build-tools/releases 进入界面如下,下载即可:

在这里插入图片描述
将下载文件拷贝到Jetson TX2 NX中,如图所示:

在这里插入图片描述
输入命令安装

./Archiconda3-0.2.3-Linux-aarch64.sh

可能无法正常安装,这里需要给文件Archiconda3-0.2.3-Linux-aarch64.sh所在文件夹权限,输入命令

sudo chmod -R 777 /home

再次运行安装命令

./Archiconda3-0.2.3-Linux-aarch64.sh

如图所示,直接按Enter

在这里插入图片描述
输入 yes ,继续按Enter

在这里插入图片描述
继续输入 yes ,按Enter,等待安装成功

在这里插入图片描述
接下来输入命令

vim ~/.bashrc

进入到如下界面,按i,在最后一行输入

export PATH=~/home/anaconda3/bin:$PATH

Esc键,然后输入:wq保存并退出

在这里插入图片描述

3 创建pytorch环境

输入命令创建pytorch环境,命令中PyTorch是虚拟环境的名字,python必须为3.6版本

conda create -n PyTorch python=3.6

如图所示:

在这里插入图片描述
虚拟环境PyTorch创建完成:

在这里插入图片描述
安装jetson-stats,方便查看jetpack版本,输入命令

sudo pip3 install jetson-stats

如图所示,正在安装

在这里插入图片描述
输入命令

sudo jtop

在这里插入图片描述
查看jetpack版本:

在这里插入图片描述
pytorch下载链接: https://forums.developer.nvidia.com/t/pytorch-for-jetson/72048 选择合适版本,直接下载即可:

在这里插入图片描述
进入创建的PyTorch虚拟环境,输入命令

conda activate PyTorch

进入之后,可以输出命令查看所安装的包

conda list

如图所示:

在这里插入图片描述
如果此时直接输入命令

pip install torch-1.8.0-cp36-cp36m-linux_aarch64.whl

将出现错误 Illegal instruction (core dumped)
该错误解决了好几天,使用了网上各种方法,最终还是没有解决,参考大佬博客: https://blog.csdn.net/lxx191079173/article/details/121231468
先输入命令,降低pip的版本

conda install pip=19.1.1

如图所示:

在这里插入图片描述
然后输入命令

pip install torch-1.8.0-cp36-cp36m-linux_aarch64.whl

仍然报错Illegal instruction (core dumped)
使用pip3,不再报错,此时欣喜若狂,以为好几天的问题终于解决了

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

如图所示,的确安装成功,仔细检查后,发现其实并没有安装在所创建的虚拟环境PyTorch中,白白高兴一场

在这里插入图片描述
继续安装torchvision,输入命令

git clone --branch v0.9.0 https://github.com/pytorch/vision torchvision

如图所示,没有成功执行

在这里插入图片描述
之前遇到过该问题,依次输入下列命令即可

git config --global --unset http.proxy
git config --global --unset https.proxy

然后输入命令

git clone --branch v0.9.0 https://github.com/pytorch/vision torchvision

成功clone,如图所示:

在这里插入图片描述
依次输入命令

cd torchvision
export BUILD_VERSION=0.9.0

如图所示:

在这里插入图片描述
继续输入命令

python3 setup.py install --user

如图所示,又报错Illegal instruction (core dumped),崩溃

在这里插入图片描述
直接手动下载torchvision-0.9.0a0+01dfa8e-cp36-cp36m-linux_aarch64.whl文件,拷贝至Jetson TX2 NX,输入命令安装

pip3 install torchvision-0.9.0a0+01dfa8e-cp36-cp36m-linux_aarch64.whl

成功安装,如图所示:

在这里插入图片描述
VSCode中测试,所选环境为之前装的python3.6环境,刚刚创建的虚拟环境PyTorch无法使用,实在木有办法啦

在这里插入图片描述

4 所尝试的比较接近成功的方法

该方法比较接近成功,有兴趣的小伙伴可以继续尝试
首先创建空的虚拟环境,即不安装python,输入命令

conda create -n PyTorch

如图所示:

在这里插入图片描述
使用如下命令安装python

conda install -c jetson-tx2 python

如图所示,如果需要安装其他包,可以参考: https://anaconda.org/jetson-tx2/repo

在这里插入图片描述
正在安装所需要的包:

在这里插入图片描述
安装完成,依次输入命令,查看虚拟环境

conda activate PyTorch
conda list

如图所示,python版本为3.6.1,pip版本为9.0.1

在这里插入图片描述
尝试安装torch,输入命令

pip install torch-1.8.0-cp36-cp36m-linux_aarch64.whl

如图所示,python版本报错

在这里插入图片描述

总结一下,本人感觉,一直报错无法正常安装pytorch的原因可能是:

  • pip版本问题,低版本应该行得通
  • python版本问题,3.6版本的python还有很多小版本
  • 其他的依赖包版本问题

希望本文对大家有帮助,上文若有不妥之处,欢迎指正

分享决定高度,学习拉开差距

;