PyTorch可以通过Jetson Nano上的APT软件包管理器进行安装,以下是安装步骤:

  1. 更新APT软件包管理器的软件源,打开终端并输入以下命令:

    $ sudo apt-get update
    

    2.安装PyTorch的依赖项,输入以下命令

 

$ sudo apt-get install libopenblas-dev libblas-dev m4 cmake cython python3-dev python3-yaml python3-setuptools

3.下载PyTorch源代码并进行编译,输入以下命令:

$ git clone --recursive https://github.com/pytorch/pytorch
$ cd pytorch
$ git submodule update --init
$ export NO_CUDA=1
$ python3 setup.py install

4.安装完成后,可以在Python中导入PyTorch进行测试,输入以下命令:

$ python3
>>> import torch
>>> print(torch.__version__)

Logo

免费领 150 小时云算力,进群参与显卡、AI PC 幸运抽奖

更多推荐