TX2入门教程软件篇-安装PyTorch(Caffe2)v1.2.0(jetpack3.3)

TX2入门教程软件篇-安装PyTorch(Caffe2)v1.2.0(jetpack3.3)

说明：

介绍如何在tx2 jetpack3.3安装PyTorch(Caffe2)v1.2.0
环境

步骤：

mkdir -p ~/dl/PyTorch
cd ~/dl/PyTorch

最大性能

$ sudo nvpmodel -m 0
$ sudo ~/jetson_clocks.sh

下载源码

$ git clone --recursive http://github.com/pytorch/pytorch
$ cd pytorch
# if you are updating an existing checkout
$ git submodule sync
$ git submodule update --init --recursive

配置环境变量

$ export USE_NCCL=0
$ export USE_DISTRIBUTED=0
$ export TORCH_CUDA_ARCH_LIST="5.3;6.2"

构建wheel 针对python2

$ sudo apt-get install python-pip cmake
$ pip install -U pip

$ sudo pip install -U setuptools
$ sudo pip install -r requirements.txt

$ pip install scikit-build --user

$ python setup.py bdist_wheel

构建wheel 针对python3

$ sudo apt-get install python3-pip cmake

$ sudo pip3 install -U setuptools
$ sudo pip3 install -r requirements.txt

$ pip3 install scikit-build --user

$ python3 setup.py bdist_wheel

编译完成后的whl文件位置

$ cd ~/dl/PyTorch/pytorch/dist
$ ls
torch-1.2.0a0+54a63e0-cp35-cp35m-linux_aarch64.whl

安装whell

# Python 2.7 (download pip wheel from above)
$ pip install torch-1.2.0a0+8554416-cp27-cp27mu-linux_aarch64.whl --user

# Python 3.6 (download pip wheel from above)
$ pip3 install torch-1.2.0a0+54a63e0-cp35-cp35m-linux_aarch64.whl --user

pip安装

 pip3 install torchvision --user

源码安装torchvision

$ git clone https://github.com/pytorch/vision
$ cd vision
$ git checkout -b v0.4.0
$ sudo python3 setup.py install

测试：

测试cuda和torch

ubuntu@AiROS:~$ python3
Python 3.5.2 (default, Jul 10 2019, 11:58:48) 
[GCC 5.4.0 20160609] on linux
Type "help", "copyright", "credits" or "license" for more information.
>>> import torch
>>> print(torch.__version__)
1.2.0a0+54a63e0
>>> print(torch.cuda.is_available())
True
>>> a = torch.cuda.FloatTensor(2)
>>> print(a)
tensor([0., 0.], device='cuda:0')
>>> b = torch.randn(2).cuda()
>>> print(b)
tensor([ 1.3568, -1.6426], device='cuda:0')
>>> c = a + b
>>> print(c)
tensor([ 1.3568, -1.6426], device='cuda:0')
>>>

测试vision

import torchvision

参考：