I do not use GPU to detect

[li33609]

I have GTX1650 but do not use it to detect but I can use CPU to detect.

detect bus.jpg and zidane.jpg

My cuda is 11.1. CPU is 4800H.

[li33609]

No box appears using GPU,but there is a detection box when using the CPU

[syn0419]

Is your problem solved? I'm having a similar issue, the results are right when I detect with GPU, but when I detect with CPU, I got too many boxes and all confidence is 1

[glenn-jocher]

@li33609 yes, sorry, this is a known issue that occurs sometimes with PyTorch on Windows.

It appears you may have environment problems. Please ensure you meet all dependency requirements if you are attempting to run YOLOv5 locally. If in doubt, create a new virtual Python 3.8 environment, clone the latest repo (code changes daily), and pip install -r requirements.txt again. We also highly recommend using one of our verified environments below.

Requirements

Python 3.8 or later with all requirements.txt dependencies installed, including torch>=1.7. To install run:

$ pip install -r requirements.txt

Environments

YOLOv5 may be run in any of the following up-to-date verified environments (with all dependencies including CUDA/CUDNN, Python and PyTorch preinstalled):

• Google Colab and Kaggle notebooks with free GPU:
• Google Cloud Deep Learning VM. See GCP Quickstart Guide
• Amazon Deep Learning AMI. See AWS Quickstart Guide
• Docker Image. See Docker Quickstart Guide

Status

If this badge is green, all YOLOv5 GitHub Actions Continuous Integration (CI) tests are passing. These tests evaluate proper operation of basic YOLOv5 functionality, including training (train.py), testing (test.py), inference (detect.py) and export (export.py) on MacOS, Windows, and Ubuntu.