【Jetson】Jetson TX2 ubuntu18.04安装PyTorch

1、准备工作

1.1 烧写TX2

使用NVIDIA的SDK Manager烧写TX2，本人烧写的版本是Jetpack4.5.1[L4T 32.5.2]

1.2 安装JTop

Jtop命令查看GPU版本信息、使用状态

sudo pip3 install jetson-stats
jtop
1
2

可以查看到TX2的计算能力版本号为为sm_6.2，CUDA版本为V10.2.89

也可以使用nvcc命令查看CUDA版本

$ nvcc --version
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2019 NVIDIA Corporation
Built on Wed_Oct_23_21:14:42_PDT_2019
Cuda compilation tools, release 10.2, V10.2.89
1
2
3
4
5

1.3 NVIDIA GPU算力版本汇总

Jetson AGX Xavier	7.2
Jetson Nano	5.3
Jetson TX2	6.2
Jetson TX1	5.3
Tegra X1	5.3

Tesla K80	3.7
Tesla K40	3.5
Tesla K20	3.5
Tesla C2075	2.0
Tesla C2050/C2070	2.0
NVIDIA A100	8.0
NVIDIA T4	7.5
NVIDIA V100	7.0
Tesla P100	6.0
Tesla P40	6.1
Tesla P4	6.1
Tes
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17