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还有很多小版本
- 其他的依赖包版本问题
希望本文对大家有帮助,上文若有不妥之处,欢迎指正
分享决定高度,学习拉开差距