一、PyTorch

1、torch下载

torch版本选择由JetPack版本决定,JetPack 是 NVIDIA 为 Jetson 系列嵌入式计算平台提供的软件开发套件(SDK),输入以下两个命令均可以查看JetPack版本,我的是5.1.2

dpkg -l | grep nvidia-jetpack

sudo apt show nvidia-jetpack

 官网不同JetPack版本对应的torch下载链接  Index of /compute/redist/jp, 我的JetPack版本是5.1.2,所以点击v512即可下载对应pytorch版本torch-2.1.0a0+41361538.nv23.06-cp38-cp38-linux_aarch64.whl

2、torch安装 

 nvidia官网在Jetson 平台安装PyTorch教程:Installing PyTorch for Jetson Platform - NVIDIA Docs

 安装过程需要联网,python版本推荐选择3.8,在终端依次输入以下命令更新和安装系统软件包:

sudo apt-get -y update
sudo apt-get -y install autoconf bc build-essential g++-8 gcc-8 clang-8 lld-8 gettext-base gfortran-8 iputils-ping libbz2-dev libc++-dev libcgal-dev libffi-dev libfreetype6-dev libhdf5-dev libjpeg-dev liblzma-dev libncurses5-dev libncursesw5-dev libpng-dev libreadline-dev libssl-dev libsqlite3-dev libxml2-dev libxslt-dev locales moreutils openssl python-openssl rsync scons python3-pip libopenblas-dev

确保终端路径为torch安装包所在路径,输入以下命令进行安装:

pip install  torch-2.1.0a0+41361538.nv23.06-cp38-cp38-linux_aarch64.whl

安装完成后发现有黄色报警,WARNING:The script isympy is installed in '/home/nvidia/.local/bin' which is not on PATH. 提示一些库的安装位置没有被添加到环境变量 PATH 中,依次输入以下命令,将该目录添加到环境变量中即可:

echo 'export PATH=$PATH:/home/nvidia/.local/bin' >> ~/.bashrc
source ~/.bashrc

二、Torchvision

torchvision版本要与torch相对应,对应关系如下:

 torch2.1对应的torchvision0.16版本下载地址:https://download.csdn.net/download/m0_73100347/90328873

在终端输入以下命令更新系统软件包:

sudo apt-get update

在终端输入以下命令升级系统软件包:

sudo apt-get upgrade

输入y后回车继续安装,出现以下界面,选择yes后回车,等待安装完成。

输入以下命令安装系统软件包和依赖:

sudo apt-get install libjpeg-dev zlib1g-dev libpython3-dev libavcodec-dev libavformat-dev libswscale-dev

确保终端路径为torchvision安装包所在路径,输入以下命令进行安装:

pip install torchvision-0.16.2+c6f3977-cp38-cp38-linux_aarch64.whl

验证安装是否成功:

三、Pymssql

我的程序需要用到数据库连接,所以还需要安装pymssql库,不需要连数据库的可以跳过这步。

确保终端路径为pymssql安装包所在路径,输入以下命令进行安装:

pip install pymssql-2.3.0-cp38-cp38-manylinux_2_17_aarch64.manylinux2014_aarch64.whl

四、yolov8相关库

删除requirements.txt中的torch和torchvision,确保终端路径为requirements.txt文件所在路径,输入以下命令进行安装:

pip install -r requirements.txt

 红色报错没有管,直接运行predict.py,程序报错如下:

输入以下命令安装sentry_sdk后成功运行yolov8程序:

pip install sentry_sdk

五、其他问题

1、python默认版本

Linux 系统中如果同时安装了 Python 2 和 Python 3,并且默认版本是 Python 2,想要改为 Python 3,先通过以下命令查看Python 3的可执行文件路径(通常为 /usr/bin/python3):

which python

然后通过以下命令将默认的 Python 版本更改为 Python 3:

sudo update-alternatives --install /usr/bin/python python /usr/bin/python3 1

2、python库下载

如果一些python库在线下载安装失败,可以先到PyPI · The Python Package Index下载对应的.whl文件到本地,然后使用以下命令进行离线安装:

pip install xxxx.whl

Logo

腾讯云面向开发者汇聚海量精品云计算使用和开发经验,营造开放的云计算技术生态圈。

更多推荐