Xavier因为架构问题, 从conda install不支持cuda...

有一个论坛:

https://forums.developer.nvidia.com/t/pytorch-for-jetson-version-1-10-now-available/72048

点开你会发现都是Cyton3.6写的. 所以Xavier似乎只能在最高Python3.6安装运行Pytorch加速, 因为 (python=3.7安装我试过不太行)

image.png

安装PyTorch

这时候查看你的Project, 需要什么版本Pytorch,对应下载, 我下载的是

https://nvidia.box.com/shared/static/cs3xn3td6sfgtene6jdvsxlr366m2dhq.whl

wget https://nvidia.box.com/shared/static/cs3xn3td6sfgtene6jdvsxlr366m2dhq.whl
sudo apt-get install python3-pip libopenblas-base libopenmpi-dev 
pip3 install Cython numpy
pip3 install torch-1.7.0-cp36-cp36m-linux_aarch64.whl

安装torchvision

有个网址, Torchvision:https://github.com/pytorch/vision

可以下载好, cd到文件夹python setup.py install, 也
当然可以直接通过conda安装torchvision
Xavier安装conda (https://www.jianshu.com/p/231036636f54)

conda install torchvision -c pytorch

最后检测一下:
Terminal输入python3

import torch
torch.is_cuda_available()