NVIDIA Jetson AGX Xavier 配置pytorch环境,anaconda,Pycharm

提示：文章写完后，目录可以自动生成，如何生成可参考右边的帮助文档

---

1.前言

花了一天配置torch环境，趁还热乎，赶紧记录下。
在实际装的过程中，主要比较坑的就是版本对应问题。还有就是是否需要配置anaconda 环境。如果需要配置虚拟环境的话，就用anaconda,因为我装的AGX4.4.1版本中，python版本为3.6.9（符合我需要），而且主要也只做一个工程项目，所以我直接在主环境中配置，也会讲解anaconda 的配置。说起来都是一把泪，当初想在虚拟环境中配置，但是就一直因为python版本不对应而无法调用torch。最后还是直接使用主环境配置成功。

2.配置版本

torch 为1.6.0
torchvision 0.7.0
AGX linux 为4.4.1版本
（其他版本的torch方法都是雷同）

3.主环境的配置

主环境的话首先是下载torch whl.文件 官网点击这里下载，可能需要翻墙
如果网络不好，我下载了几个版本，存至百度云，点击-------》下载 提取码：2233

然后按步骤安装：
1.安装torch

sudo apt-get install python3-pip libopenblas-base libopenmpi-dev
pip install Cython
pip install numpy torch-1.6.0-cp36-cp36m-linux_aarch64.whl

验证，若无报错则可以继续安装torchvision

python3
import torch

安装tochvision

sudo apt-get install libjpeg-dev zlib1g-dev
git clone --branch v0.7.0-rc2 https://github.com/pytorch/vision torchvision
cd torchvision
sudo python3 setup.py install

测试

import torch
print(torch.__version__)
print('CUDA available: