jeston nano是一块非常不错的板子,但是由于是arm架构导致部分库进行pip下载的时候会出现莫名奇妙的问题,下面我把我如何成功安装pytorch的方法写出来
1.换源
安装系统的就不多说了,相信应该都没什么问题
换源部分,由于该板子是arm架构的,所以要换特殊的源
清华源(该源好像有部分软件没有包含进去)
deb [arch=amd64,i386] http://mirrors.tuna.tsinghua.edu.cn/ubuntu/ xenial main restricted
deb [arch=amd64,i386] http://mirrors.tuna.tsinghua.edu.cn/ubuntu/ xenial-updates main restricted
deb [arch=amd64,i386] http://mirrors.tuna.tsinghua.edu.cn/ubuntu/ xenial universe
deb [arch=amd64,i386] http://mirrors.tuna.tsinghua.edu.cn/ubuntu/ xenial-updates universe
deb [arch=amd64,i386] http://mirrors.tuna.tsinghua.edu.cn/ubuntu/ xenial multiverse
deb [arch=amd64,i386] http://mirrors.tuna.tsinghua.edu.cn/ubuntu/ xenial-updates multiverse
deb [arch=amd64,i386] http://mirrors.tuna.tsinghua.edu.cn/ubuntu/ xenial-backports main restricted universe multiverse
deb [arch=amd64,i386] http://mirrors.tuna.tsinghua.edu.cn/ubuntu/ xenial-security main restricted
deb [arch=amd64,i386] http://mirrors.tuna.tsinghua.edu.cn/ubuntu/ xenial-security universe
deb [arch=amd64,i386] http://mirrors.tuna.tsinghua.edu.cn/ubuntu/ xenial-security multiverse
中科大源(软件较为齐全,作者就是用这个才安装了下文的软件)
deb https://mirrors.ustc.edu.cn/ubuntu-ports/ bionic main restricted universe multiverse
deb https://mirrors.ustc.edu.cn/ubuntu-ports/ bionic-updates main restricted universe multiverse
deb https://mirrors.ustc.edu.cn/ubuntu-ports/ bionic-backports main restricted universe multiverse
deb https://mirrors.ustc.edu.cn/ubuntu-ports/ bionic-security main restricted universe multiverse
换源之后sudo apt update一下就好了
2.安装pytorch
这部分就真的踩了点坑了。
首先我们去NVIDIA官网下载pytorch,我下的是pytorch1.6.0,网址链接在此
这里建议科学上网来下载或者在window下复制下载链接到迅雷那里下再拷过来,会快很多
下载完成后输入以下指令
sudo apt-get install python3-pip libopenblas-base libopenmpi-dev
pip3 install Cython
pip3 install numpy torch-1.6.0-cp36-cp36m-linux_aarch64.whl
作者在输入第三行指令的时候,就出现了问题,出现了非常非常多的报错内容,我这里选取几个关键的红色字报错内容
Failed building wheel for numpy
Failed cleaning build dir for numpy
Command "/usr/bin/python3 -u -c "import setuptools, tokenize;__file__='/tmp/pip-build-2x91npab/numpy/setup.py';f=getattr(tokenize, 'open', open)(__file__);code=f.read().replace('\r\n', '\n');f.close();exec(compile(code, __file__, 'exec'))" install --record /tmp/pip-k8jq0j6i-record/install-record.txt --single-version-externally-managed --compile --user --prefix=" failed with error code 1 in /tmp/pip-build-2x91npab/numpy/
我们不难看出应该是numpy安装的问题,实际上是当我们pip安装的时候,pip会帮我们把numpy也安装了,但是由于该板子神奇的架构aarm64,pip库里的numpy并不能安装成功。
所以我们用apt install来安装numpy等较为关键的库。
sudo apt-get install python3-numpy
sudo apt-get install python3-scipy
sudo apt-get install python3-pandas
sudo apt-get install python3-matplotlib
sudo apt-get install python3-sklearn
把这些装完之后,我们在进入到刚刚有torch文件的文件夹内,输入指令
sudo pip3 install numpy torch-1.6.0-cp36-cp36m-linux_aarch64.whl
应该就可以安装成功的
我们进入在终端输入python测试一下(jeston nano默认python指向是2.7,要人为修改成3.6,这个网上教程很多就不多说了)
>>> import torch
要是没问题的话就应该是安装成功了