Jetson TX2安装pytorch探坑记

第一步：安装Archiconda3

至于为什么不用anaconda3，由于不支持ARM架构，详细原因网上去查，废话不多说。到https://gitee.com/shenyinsong/archiconda3下载Archiconda3-0.2.3-Linux-aarch64.sh，并执行。

第二步：创建pytorch虚拟环境

假设archiconda3已经安装好了，注意python版本目前pytorch1.8.0只支持到3.6

$conda create -n pytorch python=3.6

输入yes，根据提示操作。

$conda activate pytorch

第三步：安装pytorch1.8.0

参照原文链接：https://blog.csdn.net/weixin_43947712/article/details/115530913

（1）wget https://nvidia.box.com/shared/static/p57jwntv436lfrd78inwl7iml6p13fzh.whl -O torch-1.8.0-cp36-cp36m-linux_aarch64.whl
（2）sudo apt-get install python3-pip libopenblas-base libopenmpi-dev
（3）pip3 install Cython
（4）pip3 install numpy torch-1.8.0-cp36-cp36m-linux_aarch64.whl

事实上，在https://forums.developer.nvidia.com/t/pytorch-for-jetson-version-1-8-0-now-available/72048可以直接下载到torch-1.8.0-cp36-cp36m-linux_aarch64.whl，如果第（1）条命令执行失败，就到官网下载。

第四步：设置环境变量

这一步如果不做，在import torch时会报Illegal instruction (core dumped)错误，torchvision也不能安装。参考链接：https://www.pythonheidong.com/blog/article/897235/03b293aa817470680901/

'''修改环境变量'''
sudo gedit /etc/profile 把 export OPENBLAS_CORETYPE=ARMV8 加入最后面一行，然后保存'''更新环境变量'''
source /etc/profile

第五步：安装torchvision0.9

官网https://github.com/pytorch/vision目前是0.9.1的，据说不能用，只能用下面方法安装。

（1）sudo apt-get install libjpeg-dev zlib1g-dev libpython3-dev libavcodec-dev libavformat-dev libswscale-dev
（2）git clone --branch v0.9.0 https://github.com/pytorch/vision torchvision # see below for version of torchvision to download
（3）cd torchvision
（4）export BUILD_VERSION=0.9.0 # where 0.x.0 is the torchvision version
（5）python3 setup.py install --user

如果报错，只需多试几次就行了。

Jetson TX2安装pytorch探坑记相关推荐

ROS学习08-NVIDIA JETSON TX2安装配置激光雷达-思岚RPLIDAR S1(在Rviz上显示点位图)
ROS学习08-NVIDIA JETSON TX2安装配置激光雷达-思岚RPLIDAR S1(在Rviz上显示点位图) 在前一节我们创建了ROS工作空间,这一节开始要给我们的机器人装一些感官了,这里我 ...
jetson tx2 安装cuda 10.2或者cuda 9全过程
目录 1. 检查cuda 2. 提前安装一些必要软件 3. 安装jetson使用的cuda和cudnn 基础环境: jetpack 4.4,python3.6, ubuntu18.04 1. 检查cu ...
TX2安装pytorch+TensorRT+yolov5实现实时检测
已完成的环境配置: TX2刷机后,完成了opencv4.5.1的编译:Ubuntu18.04安装opencv4.5.1+contrib 支持cuda加速(附带编译好的opencv4.5.1及缺失文件) ...
【ARM-Linux开发】【CUDA开发】【深度学习与神经网络】Jetson Tx2安装相关之二
前言本文主要参考YouTube视频,<JetPack 3.0 - NVIDIA Jetson TX2> 视频链接如下: https://www.youtube.com/watch?v= ...
Jetson Nano安装pytorch 基于torch1.6和torchvision0.7
需要注意的是,博主使用的是win10主机,通过局域网连接的jetson nano, 其中jetson nano的预制CUDA版本为10.2 Jetpack 4.1.1 分别执行以下命令,即可查看自己的 ...
Jetson TX2 安装 D435i ROS驱动
目录 0 概述 1 通过ROS功能包安装 2 通过Realsense官方途径安装 2.1 安装librealsense2 2.1.1使用Debian包安装 2.1.2 从源码编译安装 2.2 安装li ...
Jetson TX1安装 Pytorch
大体流程 1. 重装系统先重装了系统,避免混乱的旧系统所带来的不必要麻烦. 安装依赖包 sudo apt-get install python3-pip cmake pip3 install num ...
Jetson nano安装Pytorch详解
一.官网下载安装包 NVIDIA官网:https://forums.developer.nvidia.com/t/pytorch-for-jetson-version-1-7-0-now-availa ...
NVIDIA Jetson TX2 安装 Astra相机的ros驱动源码错误总结
安装依赖确保 $ echo $ROS_DISTRO melodic 然后安装 sudo apt install ros-$ROS_DISTRO-rgbd-launch \ ros-$ROS_DIST ...

Jetson TX2安装pytorch探坑记

Jetson TX2安装pytorch探坑记相关推荐

最新文章

热门文章