Category Archives: Python

ultralytics的yolo如何只安装cpu版本？

--extra-index-url https://download.pytorch.org/whl/cpu
ultralytics
requirements中添加如上设定[......]

新版Python不再建议直接pip安装了，而是要使用虚拟环境
sudo apt install -y python3-dev python3-virtualenv
创建虚拟环境
virtualenv newenv
source newenv/bin/activate

# 取消激活
deactivate
[......]

import cv2
import numpy as np
import onnxruntime as ort

def load_model(model_path):
"""加载ONNX模型"""
session = ort.InferenceSession(model_path)
return session

def preprocess_image(image_path):
image = cv2.imread(image_path)[......]

sudo pip3 install -i https://pypi.tuna.tsinghua.edu.cn/simple xxx-package
[......]

安装
sudo pip3 install nuitka
编译成可执行程序
python3 -m nuitka --standalone --follow-imports ./test.py
会生成一个test.dist的目录，把整个目录拷贝上去，就可以执行了。

需要说明的是，现在nuitka不支持静态链接，所以如果glibc相关库对不上的话，也是没法执行的，建议安装个同版本的系统(docker即可)，去编译下就好。[......]