【问题解决】Jetson nano 安装pytorch使用GPU推理

一. 问题描述

安装 yolov8 后只调用cpu推理图片

二. 解决步骤

2.1 在推理环境下，执行下面命令卸载pytorch

pip uninstall torch torchtext torchaudio

2.2 下载PyTorch的依赖:

sudo apt-get -y update; 
sudo apt-get -y install libopenblas-dev;

###2.3 下载pytorch

export TORCH_INSTALL=https://developer.download.nvidia.cn/compute/redist/jp/v511/pytorch/torch-2.0.0+nv23.05-cp38-cp38-linux_aarch64.whl

下载pytorch命令依据当前系统的jetpack版本
https://developer.download.nvidia.com/compute/redist/jp/v$JP_VERSION/pytorch/$PYT_VERSION

python3 -m pip install --upgrade pip; python3 -m pip install numpy==’1.26.1’; python3 -m pip install --no-cache $TORCH_INSTALL

三. 测试

3.1 测试pytorch-gpu是否可用

import torch
print(torch.cuda.is_available())

3.2 查看conda 包

输入conda list 看到torch 版本为2.0.0+nv23.5。

四. 参考

nvidia官网：https://docs.nvidia.com/deeplearning/frameworks/install-pytorch-jetson-platform/index.html