基于目前最新的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