Advertisement

Jetson tx2 上源码安装 pytorch1.0.0(血的历程)

阅读量:

Jetson tx2 上源码安装 pytorch1.0.0(真. 血泪史)

本文以Python3.5的安装流程作为实例进行说明。在开始安装操作前,需注意以下几点:

重点一:平台及cuda cudnn的安装问题

复制代码
    Jetson TX2平台版本:Jetpack 3.3, cuda 9.0.252, cudnn7.1.5, TensorRT4.0.2, python2.7/python3.5
    
    系统内核:tegra-ubuntu 4.4.38-tegra aarch64
    
    Linux系统版本:Ubuntu16.04,cmake 3.15.6 (TX2刷机完原始的cmake是3.5.1版本,由于后面自己捣鼓的时候说最好安装3.9.0以上版本cmake,所以我就直接升级到新版本了)
    
    
      
      
      
      
      
    

【在进行pytorch源码安装过程中,需要用到cuda和cudnn相关组件,首先需要确认Jetson TX2设备上的cuda和cudnn是否是通过jetpack进行安装的,若并非如此,则需格外注意。Jetson TX2所搭载的CPU基于ARM架构,因此所安装的cuda与cudnn版本必须为适用于ARM平台的aarch64架构。关于在J

全部评论 (0)

还没有任何评论哟~