Jetson Nano 上安装 tensorflow2.1 和 pytorch1.4

基于目前最新的JetPack 4.3的镜像。
开机配置什么的请参考博客

第一步：换pip源

1.根目录创建.pip文件：mkdir ~/.pip
2.创建文件pip.conf：vim .pip/pip.conf
3.点击“i”键，进入编辑模式，复制信息：

sudo mkdir ~/.pip
cd ~/.pip
sudo touch pip.conf
sudo nano ~/.pip/pip.conf
#写入下面这些配置
[global] 
index-url = http://pypi.douban.com/simple 
[install] 
trusted-host=pypi.douban.com

第二步：安装相关依赖

sudo apt-get install libhdf5-serial-dev hdf5-tools libhdf5-dev zlib1g-dev zip libjpeg8-dev
sudo apt-get install python3-pip
sudo pip3 install -U pip
sudo pip3 install -U numpy grpcio absl-py py-cpuinfo psutil portpicker six mock requests gast h5py astor termcolor protobuf keras-applications keras-preprocessing wrapt google-pasta

第三步：安装tensorflow2.1:

$ sudo pip3 install --pre --extra-index-url https://developer.download.nvidia.com/compute/redist/jp/v43 tensorflow==2.1.0+nv20.3

其中在安装2.1版本的时候会出现scipy 1.4.1编译出现问题导致无法安装，
不过这个scipy1.4.1我已经编译好了。

链接：https://pan.baidu.com/s/1VdsNq-ETnIlN5mG8rJXDKw
提取码：6cnb

建议先安装scipy 1.4.1 之后再安装 tensorflow2.1 这样就不会报错了

第四步：顺便安装个keras:

sudo pip3 install keras

第五步：安装pytorch1.4:

wget https://nvidia.box.com/shared/static/ncgzus5o23uck9i5oth2n8n06k340l6k.whl -O torch-1.4.0-cp36-cp36m-linux_aarch64.whl
sudo apt-get install python3-pip libopenblas-base
sudo pip3 install Cython
sudo pip3 install numpy torch-1.4.0-cp36-cp36m-linux_aarch64.whl