Jetson Nano 安装配置conda、pytorch1.7.0

jetson nano是自带CUDA的,按照官方的教程文档可以逐步配置好ubuntu镜像、python和tensorflow的。这两天突然想配置上conda和pytorch,于是浪费了两天的时间,终于在nano上跑通,能够训练模型了,现将过程记录,方便大家用到。

第一步,安装conda

jetson nano是aarch64架构的,常用的windows,mac和linux都不适用,查看命令如下:

(base) my_nano@jetson-nano:~$ uname -a
Linux jetson-nano 4.9.140-tegra #1 SMP PREEMPT Tue Oct 27 21:02:37 PDT 2020 aarch64 aarch64 aarch64 GNU/Linux

安装conda 首先想到的肯定是anaconda,但是conda太大了,于是放弃(我的jetson nano上仅用了32G的TF卡)。

Jetson Nano 安装配置conda、pytorch1.7.0

于是尝试miniconda, 恰好miniconda有aarch64架构版本,但是安装失败了(安装成功,但是在用conda创建环境时提示:illegel instruction(core dumped) ),于是再次放弃。

Jetson Nano 安装配置conda、pytorch1.7.0

于是只能采用miniconda的替代品–miniforge。miniforge的下载地址:

miniforge下载链接GitHub – conda-forge/miniforge: A conda-forge distribution.,打开链接之后,点击aarch64版本即可下载安装包。

Jetson Nano 安装配置conda、pytorch1.7.0

进入包含miniforge安装包的目录,右键open终端后输入命令安装:

bash Miniforge3-Linux-aarch64.sh -b

安装结束后按如下命令初始化conda ,初始化后重新打开终端。

~/miniforge3/bin/conda init

输入命令:conda -V查看版本:

(base) my_nano@jetson-nano:~/miniforge3/bin$ conda -V
conda 4.10.3

第2步,我们创建一个环境,python用3.6版本。

conda create --name jetson_nano python=3.6 -y

第3步,安装pytorch,jetson nano是安装了CUDA的,可以用命令查看:

(base) my_nano@jetson-nano:~$ nvcc -V
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2019 NVIDIA Corporation
Built on Wed_Oct_23_21:14:42_PDT_2019
Cuda compilation tools, release 10.2, V10.2.89

输入命令查看当前conda的环境,如果没有意外,应该有两个,一个是base,另一个是jetson_nano(我的是三个,没有定义jetson_nano)

(base) my_nano@jetson-nano:~/miniforge3/bin$ conda env list
conda environments:
#
base                  *  /home/my_nano/miniforge3
d2l                      /home/my_nano/miniforge3/envs/d2l
d2l-36                   /home/my_nano/miniforge3/envs/d2l-36

pytorch可以安装在base环境下,也可以选择安装在某一个环境下,如果要指定安装在某个虚拟环境,请输入conda activate (比如是:jetson_nano或者d2l等等自定义的名字,注意左侧的括号代表的就是你要安装的pytorch的环境,且python版本是上面指定的V3.6,为什么一定是3.6后面说。。。)

(base) my_nano@jetson-nano:~/miniforge3/bin$ conda activate d2l-36
(d2l-36) my_nano@jetson-nano:~/miniforge3/bin$

查看JetPack的版本,在终端输入jtop:

Jetson Nano 安装配置conda、pytorch1.7.0

jetson安装pytorch需要从官网下载安装包,当然,你需要fanqiang,所以呢,链接地址如下::

https://pan.baidu.com/s/1134QgCsZyoDL3E6O14i__w​​​​​​​​​​​​​​,提取码: g5mq

下载版本高于V1.6.0,我安装的是V1.7.0:

torch-1.7.0-cp36-cp36m-linux_aarch64,其中cp36m是指python3.6.

vision_0.8的链接地址如下:

链接:百度网盘 请输入提取码 提取码:wts3

在需要安装的conda环境下执行命令:

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

结束安装后可验证:

(d2l-36) my_nano@jetson-nano:~/下载$ ipython
Python 3.6.9 (default, Jan 26 2021, 15:33:00)
Type 'copyright', 'credits' or 'license' for more information
IPython 7.16.1 -- An enhanced Interactive Python. Type '?' for help.

In [1]: import torch

In [2]: import torchvision

In [3]: print(torch.__version__)
1.7.0

In [4]: print(torchvision.__version__)
0.10.0

安装成功。

我的公众号:

Jetson Nano 安装配置conda、pytorch1.7.0

Original: https://blog.csdn.net/weixin_41747193/article/details/119303645
Author: 王三思
Title: Jetson Nano 安装配置conda、pytorch1.7.0

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

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

(0)

大家都在看

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