更新于:2019-8-15
参考了很多博主的安装记录,总结一下更方便一些。
1.装机教程:
直接参考这个博主的就可以了https://blog.csdn.net/qq_33869371/article/details/89208667 ,这个还附带了安装TensorFlow的教程。建议用原装系统,里面有很多配置不用从头来。而且基于arm64构架的大部分Ubuntu软件安装不了。
2.安装pytorch:
难点是需要:
Python2
wget https://nvidia.box.com/shared/static/o8teczquxgul2vjukwd4p77c6869xmri.whl -O torch-1.1.0-cp27-cp27mu-linux_aarch64.whl
pip install torch-1.1.0-cp27-cp27mu-linux_aarch64.whl
Python3
wget https://nvidia.box.com/shared/static/j2dn48btaxosqp0zremqqm8pjelriyvs.whl -O torch-1.1.0-cp36-cp36m-linux_aarch64.whl
pip3 install numpy torch-1.1.0-cp36-cp36m-linux_aarch64.whl
当然要给百度云:
python3的安装包链接:https://pan.baidu.com/s/1zCfAPXhQh59-0rrZLWbUpg 提取码:1xe7
python2的安装包链接:https://pan.baidu.com/s/19CU4pUwbP6I0zw9C7kQhBQ 密码:0g9i
我只装了python3版本的,然后发现python2版本装不了了,也就算了。建议只装一个。
张python3版本时会发现系统没有安装pip3,需要安装一下相关的依赖库和工具:
sudo apt-get install libhdf5-serial-dev hdf5-tools zlib1g-dev zip libjpeg8-dev libhdf5-dev python3-pip
因为系统里面默认是没有python3-pip的,所以需要自己安装。
装完后测试pytorch:
import torch
print(torch.__version__)
print('CUDA available: ' + str(torch.cuda.is_available()))
a = torch.cuda.FloatTensor(2).zero_()
print('Tensor a = ' + str(a))
b = torch.randn(2).cuda()
print('Tensor b = ' + str(b))
c = a + b
print('Tensor c = ' + str(c))
torchvision也应该安装一下
# 安装一些必要的包
sudo apt-get install libjpeg-dev zlib1g-dev
# 下载torchvision
git clone -b v0.3.0 https://github.com/pytorch/vision torchvision
# 安装torchvision
cd torchvision
sudo python3 setup.py install
测试torchvision:
import torchvision
print(torchvision.__version__)
参考博主:https://blog.csdn.net/beckhans/article/details/91386429
官网:https://devtalk.nvidia.com/default/topic/1049071/jetson-nano/pytorch-for-jetson-nano/
3。安装opencv,TensorFlow,Keras
一开始我发现nano的python2是自带opencv包的,python3没有,需要在上一步把python3的pip3等包更新后才会在python3中可以import cv2.
安装TensorFlow,Keras和其他各种包参考:
https://blog.csdn.net/beckhans/article/details/89138876
https://blog.csdn.net/beckhans/article/details/89146881
这个博主写的挺全的,很nice