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)
还没有任何评论哟~
