pytorch怎么正确安装到jetson

最新推荐文章于 2024-08-07 10:53:45 发布

发愤图强想做全栈的小陈

最新推荐文章于 2024-08-07 10:53:45 发布

阅读量428

点赞数

分类专栏： jetson 文章标签： pytorch ubuntu linux

本文链接：https://blog.csdn.net/qq_34837666/article/details/129092808

版权

3 篇文章 0 订阅

订阅专栏

本文详细介绍了如何在JetsonNano设备上通过APT软件包管理器来安装PyTorch。首先更新软件源，然后安装必要的依赖项，接着下载并编译PyTorch源代码，最后在Python环境中测试PyTorch是否成功安装。

摘要由CSDN通过智能技术生成

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

更新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__)

关注

专栏目录