一、Anaconda
下载安装包
首先,到anaconda官网下载安装包,注意不是下载平常linux系统所用的安装包,而是下载ARM架构的安装包,若使用其他的版本会提示版本不符合无法安装。
安装
然后将下载好的安装包拷贝到Jetson nx里,使用命令bash Anaconda3-2021.05-Linux-aarch64.sh
进行安装,不懂怎么安装的小伙伴看此链接:https://blog.csdn.net/qingfengxd1/article/details/109334199
修改配置
若不懂如何修改**~/.bashrc**的小伙伴参考此链接:https://blog.csdn.net/hehedadaq/article/details/115130945,
步骤简单来说就是:
vi ~/.bashrc
按一下 “ i ”,此时是插入模式,可输入内容,去到最后一行,输入你要输入的内容:export …
输入完成后想退出并保存则:按一下ESC键,输入“ :wq! ”,回车,输入
source ~/.bashrc
构建虚拟环境
成功安装anaconda后,进行基本的构建虚拟环境操作,
①添加清华源:https://blog.csdn.net/Rocky6688/article/details/104082982,运行前三行命令就够了,也可以自行搜索添加豆瓣源
②创建虚拟环境:
我这里用的是python3.6,因为后面重头戏torch,官方给的例子是3.6,以防又有bug我就直接用的3.6
conda create -n gg python=3.6
conda activate gg
二、torch与torchvision
pytorch官网提供的离线包只有CPU版本的,只需要CPU的小伙伴可自行下载,直接Ctrl+F搜索aarch64就有了,但是!!!GPU版本的就不是用这个链接的安装包了
GPU版本的torch离线包链接地址
想用GPU的小伙伴,看NVIDIA官方给的这个链接:https://forums.developer.nvidia.com/t/pytorch-for-jetson-version-1-9-0-now-available/72048,跟着官网的步骤走就行
我选择的是pytorchv1.8.0,因为官网给出的对应关系里没有说明pytorch1.9.0和torchvision哪个版本对应,本人为避免又出现bug,就选择了最为稳妥的1.8.0版本
可以看到官网对应的是python3.6,所以之前构建虚拟环境时我使用的是3.6,各位小伙伴如果安装了其他版本的python,我也不保证是否可行,你们可以试试
安装torch
点开Installation就有安装命令了,一行一行来即可
安装torchvision
torch安装完成了就到torchvision了
看图片里我标注的注意事项,pillow的安装直接pip install pillow就行,总之其他模块安装时若报错,大家可以尝试灵活处理,最主要是torch和torchvision没错就没什么大问题
祝各位也成功用GPU起飞
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系:hwhale#tublm.com(使用前将#替换为@)