语义分割学习笔记2——Jetson Xavier安装pytorch

╰+攻爆jí腚メ 2022-10-05 05:44 137阅读 0赞

### 语义分割学习笔记2——Jetson Xavier安装pytorch ###

*  一、whl文件安装pytorch
 *  二、安装torchvision
 *  三、测试是否安装成功

运行语义分割的代码，需要先安装pytorch，下面是我在Xavier上安装语义分割的步骤。  
我的环境说明：

<table> 
 <thead> 
  <tr> 
   <th>名称</th> 
   <th>版本</th> 
  </tr> 
 </thead> 
 <tbody> 
  <tr> 
   <td><strong>硬件</strong>：</td> 
   <td><strong>Nvidia Jetson Xavier AGX</strong></td> 
  </tr> 
  <tr> 
   <td><strong>CUDA版本</strong>：</td> 
   <td>10.2</td> 
  </tr> 
  <tr> 
   <td><strong>Jetpack版本</strong>：</td> 
   <td>JP 4.5</td> 
  </tr> 
  <tr> 
   <td><strong>python版本</strong>：</td> 
   <td>python2.7 python3.6</td> 
  </tr> 
  <tr> 
   <td><strong>deepstream-app version</strong></td> 
   <td>5.0.0</td> 
  </tr> 
  <tr> 
   <td><strong>TensorRT Version</strong>:</td> 
   <td>7.1</td> 
  </tr> 
  <tr> 
   <td><strong>cuDNN Version</strong>:</td> 
   <td>8.0</td> 
  </tr> 
 </tbody> 
</table>

# 一、whl文件安装pytorch #

在英伟达官方提供的[Jetson Zoo][]处下载你系统对应的英伟达编译好的arm版本，这里和平常linux系统下安装的whl包不同，Jetson Xavier是arm环境不是x86。  
例如我下载的文件是

然后按照下面四个步骤安装pytorch，这里注意，不是直接运行  
**pip install torch-1.6.0-cp36-cp36m-linux\_aarch64.whl**  
而是需要运行下面四个步骤才能成功。

--------------------

安装 OpenBLAS and OpenMPI

sudo apt-get install libopenblas-base libopenmpi-dev

sudo apt-get install python3-pip
    pip3 install Cython
    pip3 install numpy torch-1.6.0-cp36-cp36m-linux_aarch64.whl

![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L21hb19odWlfZmVp_size_16_color_FFFFFF_t_70]

测试是否运行成功

python3
    import torch

![在这里插入图片描述][20210607095232325.png]

# 二、安装torchvision #

pytorch和torchvision版本对应关系  
以下版本截止2021年3月24日  
![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L21hb19odWlfZmVp_size_16_color_FFFFFF_t_70 1]  
pytorch1.6.0对应的torchvision版本为0.7.0

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

![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L21hb19odWlfZmVp_size_16_color_FFFFFF_t_70 2]

参考：  
[https://blog.csdn.net/u013595260/article/details/107180928/][https_blog.csdn.net_u013595260_article_details_107180928]

# 三、测试是否安装成功 #

import torch
    import torchvision
    
    print(torch.__version__)
    print(torch.version.cuda)
    print(torchvision.__version__)

运行

python3 test_pytorch.py

![在这里插入图片描述][20210607095332798.png]

大功告成!

[Jetson Zoo]: https://elinux.org/Jetson_Zoo#PyTorch_.28Caffe2.29
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L21hb19odWlfZmVp_size_16_color_FFFFFF_t_70]: /images/20221005/b2704a1bfd2a4fe8837d865cc941c8f5.png
[20210607095232325.png]: /images/20221005/5d5e5ee949404056ae29777e1f26e37f.png
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L21hb19odWlfZmVp_size_16_color_FFFFFF_t_70 1]: /images/20221005/b2d9fec1b8424fb8bafcaa671f91419b.png
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L21hb19odWlfZmVp_size_16_color_FFFFFF_t_70 2]: /images/20221005/0be47d05cd074ec28b0ea0765643b5a2.png
[https_blog.csdn.net_u013595260_article_details_107180928]: https://blog.csdn.net/u013595260/article/details/107180928/
[20210607095332798.png]: /images/20221005/6df94de03d984334aeb9faf141bf2228.png