1. 首页
  2. 人工智能
  3. 深度学习
  4. nvidia jetson上的PyTorch安装教程

nvidia jetson上的PyTorch安装教程

上传者: 2023-05-22 21:08:56上传 ZIP文件 118.99MB 热度 25次

在nvidia jetson设备上安装PyTorch非常简单。我们只需从官方网站下载适用于Linux aarch64架构的torch-1.9.0a0+torchvision-0.10.0-cp36m-linux_aarch64文件并进行安装即可。首先,我们需要确保已安装Python 3.6及以上版本和pip包管理器。然后,我们可以使用以下命令安装PyTorch:

pip3 install torch-1.9.0a0+torchvision-0.10.0-cp36m-linux_aarch64.whl

完成后,我们还需要验证安装是否成功。我们可以创建一个简单的Python脚本来导入torch和torchvision库并打印版本号:

import torch

import torchvision

print(torch.version)

print(torchvision.version)

如果能够正常运行,那么我们已经成功安装了PyTorch!

用户评论