Jetson 系列开发板(NX/AGX /Nano)搭建pytorch-gpu环境

jupiter
2022-05-16 / 4 评论 / 1,725 阅读 / 正在检测是否收录...
温馨提示:
本文最后更新于2022年05月16日,已超过649天没有更新,若内容或图片失效,请留言反馈。
提醒:Jetson Xavier NX 用不了 nvidia-smi 命令

Jetson Xavier NX

0.查看JetPack版本信息

sudo apt-cache show nvidia-jetpack
Package: nvidia-jetpack
Version: 4.6-b199
Architecture: arm64
Maintainer: NVIDIA Corporation
Installed-Size: 194
  • 后面选择安装版本的时候需要根据JetPack版本信息选择版本

1.安装miniconda

wget https://github.com/Archiconda/build-tools/releases/download/0.2.3/Archiconda3-0.2.3-Linux-aarch64.sh
bash Archiconda3-0.2.3-Linux-aarch64.sh
  • 然后创建自己的虚拟环境即可
conda create -n base-jupiter python=3.6

2.安装pytorch-gpu

wget https://nvidia.box.com/shared/static/fjtbno0vpo676a25cgvuqc1wty0fkkg6.whl -O torch-1.10.0-cp36-cp36m-linux_aarch64.whl
pip install numpy Cpython
pip install torch-1.10.0-cp36-cp36m-linux_aarch64.whl

遇到问题及解决办法

  • ImportError: libopenblas.so.0: cannot open shared object file: No such file or directory
sudo apt-get install libopenblas-dev
  • OSError: libmpi_cxx.so.20: cannot open shared object file: No such file or directory
sudo apt-get install libopenmpi-dev 
  • import torch 出现 Illegal instruction (core dumped)
vim ~/.bashrc
# 把以下内容加入到末尾扩充环境变量
export OPENBLAS_CORETYPE=ARMV8
source ~/.bashrc

3.安装torchvision

torchtorchvisionpython
main / nightlymain / nightly>=3.7, <=3.10
1.11.00.12.0>=3.7, <=3.10
1.10.20.11.3>=3.6, <=3.9
1.10.10.11.2>=3.6, <=3.9
1.10.00.11.1>=3.6, <=3.9
1.9.10.10.1>=3.6, <=3.9
1.9.00.10.0>=3.6, <=3.9
1.8.20.9.2>=3.6, <=3.9
git clone -b v0.11.1 https://github.com/pytorch/vision.git vision-0.11.1
cd vision-0.11.1
export BUILD_VERSION=0.11.1
python setup.py install

4.效果测试

(base-jupiter) nvidia@nx:~$ python
Python 3.6.15 | packaged by conda-forge | (default, Dec  3 2021, 19:12:04)
[GCC 9.4.0] on linux
Type "help", "copyright", "credits" or "license" for more information.
>>> import torch
>>> torch.cuda.is_available()
True

参考资料

  1. 查看Jetson系列产品JetPack的版本信息
  2. NVIDIA Jetson Xavier NX搭建pytorch gpu环境(超详细)
  3. NVIDIA JETSONTX2 安装 pytorch 出现错误:import torch 出现 Illegal instruction(core dumped)
  4. ImportError: libopenblas.so.0: cannot open shared object file: No such file or directory
  5. Jetson AGX Xavier安装Archiconda虚拟环境管理器与在虚拟环境中调用opencv
  6. Jetson AGX Xavier安装torch、torchvision且成功运行yolov5算法
  7. https://github.com/pytorch/vision
0

评论 (4)

打卡
取消
  1. 头像
    郭老二
    Windows 10 · Google Chrome

    NVIDIA已经不能下载,大神可以发下 torch-1.10.0-cp36-cp36m-linux_aarch64.whl 到邮箱吗?
    gwlbl@163.com

    回复
  2. 头像
    ise
    Ubuntu · FireFox

    大佬,既然torch有gpu版的,那么torchision是不是也得有gpu版的阿。
    在download.pytorch这里,我注意到一些torchision前面也是有cu xxx前缀的

    回复
    1. 头像
      jupiter 作者
      Windows 10 · Google Chrome
      @ ise

      是的,但是博客里面写的是编译安装的,github里面有写到:

      By default, GPU support is built if CUDA is found and torch.cuda.is_available() is true. It's possible to force building GPU support by setting FORCE_CUDA=1 environment variable, which is useful when building a docker image.

      只要设备是支持GPU的,编译出来的版本默认就是支持cuda/gpu的。

      回复
      1. 头像
        ise
        Windows 7 · Google Chrome
        @ jupiter

        非常感谢回复,刚刚我也发现了这一点,所以正在git那个torchvision的源码

        回复