Comments (7)
Hi @Neizvestnyj,
Thank you for confirming that the issue with yolo check
is minor. It's good to know that your GPUs are being recognized during training. If you encounter any further issues or have additional questions, please feel free to reach out. We're here to help you get the most out of your setup! 🚀
Happy training! 😊
from ultralytics.
👋 Hello @Neizvestnyj, thank you for your interest in Ultralytics YOLOv8 🚀! We recommend a visit to the Docs for new users where you can find many Python and CLI usage examples and where many of the most common questions may already be answered.
If this is a 🐛 Bug Report, please provide a minimum reproducible example to help us debug it.
If this is a custom training ❓ Question, please provide as much information as possible, including dataset image examples and training logs, and verify you are following our Tips for Best Training Results.
Join the vibrant Ultralytics Discord 🎧 community for real-time conversations and collaborations. This platform offers a perfect space to inquire, showcase your work, and connect with fellow Ultralytics users.
Install
Pip install the ultralytics
package including all requirements in a Python>=3.8 environment with PyTorch>=1.8.
pip install ultralytics
Environments
YOLOv8 may be run in any of the following up-to-date verified environments (with all dependencies including CUDA/CUDNN, Python and PyTorch preinstalled):
- 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 Ultralytics CI tests are currently passing. CI tests verify correct operation of all YOLOv8 Modes and Tasks on macOS, Windows, and Ubuntu every 24 hours and on every commit.
from ultralytics.
@Neizvestnyj hi there,
Thank you for reaching out and providing detailed information about your setup. It's great to see that you've already checked the availability of your GPUs with torch.cuda.is_available()
and torch.cuda.device_count()
.
To help us better understand and resolve the issue, could you please provide a minimal reproducible example? This will allow us to replicate the problem on our end. You can find guidance on creating a minimal reproducible example here: Minimum Reproducible Example.
In the meantime, please ensure that you are using the latest versions of all relevant packages, including Ultralytics YOLOv8 and PyTorch. Sometimes, updating to the latest versions can resolve unexpected issues.
Additionally, you might want to verify that the GPUs are correctly specified in your training command. Here's an example of how to set the device argument for multi-GPU training:
from ultralytics import YOLO
# Load a model
model = YOLO('yolov8n.pt')
# Train the model on multiple GPUs
results = model.train(data='datasets/data.yaml', epochs=100, imgsz=640, batch=16, device=[0, 1, 2, 3])
If the issue persists, please share any error messages or logs that you encounter. This will help us diagnose the problem more effectively.
Looking forward to your response!
from ultralytics.
The thing is that yolo check
in the console outputs only CUDA 0, so I think it doesn't matter which example. But if you want to know, everything is specified as you have.
from ultralytics.
It's quite strange, but the yolo check
does not shows that there are 4 GPUs. I added example.
But when I call model.train
I see this output:
Ultralytics YOLOv8.2.48 🚀 Python-3.10.0 torch-2.0.0+cu118 CUDA:0 (NVIDIA A2, 14955MiB)
CUDA:1 (NVIDIA A2, 14955MiB)
CUDA:2 (NVIDIA A2, 14955MiB)
CUDA:3 (NVIDIA A2, 14955MiB)
from ultralytics.
Hi @Neizvestnyj,
Thank you for providing the additional details and example code. It's great to see that your GPUs are recognized during the training process, even though yolo check
doesn't list all of them.
To further investigate this issue, here are a few steps you can take:
-
Verify Package Versions: Ensure you are using the latest versions of Ultralytics YOLOv8 and PyTorch. Sometimes, updating to the latest versions can resolve unexpected issues.
-
Environment Check: Double-check your environment setup to ensure that all dependencies are correctly installed and there are no conflicts. You can use the following command to list all installed packages and their versions:
pip list
-
GPU Visibility: Since your GPUs are recognized during training, it might be a minor issue with the
yolo check
command. You can manually verify GPU visibility using the following code snippet:import torch print(f"CUDA available: {torch.cuda.is_available()}") print(f"Number of GPUs: {torch.cuda.device_count()}") for i in range(torch.cuda.device_count()): print(f"GPU {i}: {torch.cuda.get_device_name(i)}")
-
Training Command: When running your training command, ensure you explicitly specify the GPUs you want to use. Here's an example:
from ultralytics import YOLO # Load a model model = YOLO('yolov8n.pt') # Train the model on multiple GPUs results = model.train(data='coco8.yaml', epochs=100, imgsz=640, batch=16, device=[0, 1, 2, 3])
If the issue persists, please let us know, and we can further investigate. Your feedback is valuable, and we're here to help you get the most out of your training setup.
from ultralytics.
Hi @Neizvestnyj,
Thank you for providing the additional details and example code. It's great to see that your GPUs are recognized during the training process, even though
yolo check
doesn't list all of them.To further investigate this issue, here are a few steps you can take:
- Verify Package Versions: Ensure you are using the latest versions of Ultralytics YOLOv8 and PyTorch. Sometimes, updating to the latest versions can resolve unexpected issues.
- Environment Check: Double-check your environment setup to ensure that all dependencies are correctly installed and there are no conflicts. You can use the following command to list all installed packages and their versions:
pip list- GPU Visibility: Since your GPUs are recognized during training, it might be a minor issue with the
yolo check
command. You can manually verify GPU visibility using the following code snippet:import torch print(f"CUDA available: {torch.cuda.is_available()}") print(f"Number of GPUs: {torch.cuda.device_count()}") for i in range(torch.cuda.device_count()): print(f"GPU {i}: {torch.cuda.get_device_name(i)}")- Training Command: When running your training command, ensure you explicitly specify the GPUs you want to use. Here's an example:
from ultralytics import YOLO # Load a model model = YOLO('yolov8n.pt') # Train the model on multiple GPUs results = model.train(data='coco8.yaml', epochs=100, imgsz=640, batch=16, device=[0, 1, 2, 3])If the issue persists, please let us know, and we can further investigate. Your feedback is valuable, and we're here to help you get the most out of your training setup.
Hi, thank you. Its minor issue with the yolo check
.
from ultralytics.
Related Issues (20)
- Accuracy drop in ONNX and two class segmentation model HOT 10
- Triton reasoning onnx model takes category names
- Rtsp connection error robustness handling HOT 37
- After installing the ultralytics library, GPU usage decreased by half and Ksampler time doubled HOT 3
- How to decode the keypoints output by YOLO-pose? HOT 5
- Unexpected change in detection probability as image is translated HOT 1
- ultralytics masks to yolo seg convertor not working HOT 11
- Features for Individual Objects HOT 1
- customize dataset class and data loader HOT 1
- YOLO v8 doesn't use albumentations in classification task HOT 1
- labels_to_class_weights & labels_to_image_weights HOT 1
- Save the detected annotations file to the runs file HOT 1
- RTSP Stream breaks using YOLOv8 live detection HOT 9
- NotImplementedError HOT 3
- Seed setting HOT 2
- OverflowError: Python int too large to convert to C long HOT 12
- Questions regarding 'batch' and 'workers' arguments. HOT 2
- Memory leaks in your examples. HOT 4
- Non-Ascii Characters HOT 4
- False positives problemm HOT 1
Recommend Projects
-
React
A declarative, efficient, and flexible JavaScript library for building user interfaces.
-
Vue.js
🖖 Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.
-
Typescript
TypeScript is a superset of JavaScript that compiles to clean JavaScript output.
-
TensorFlow
An Open Source Machine Learning Framework for Everyone
-
Django
The Web framework for perfectionists with deadlines.
-
Laravel
A PHP framework for web artisans
-
D3
Bring data to life with SVG, Canvas and HTML. 📊📈🎉
-
Recommend Topics
-
javascript
JavaScript (JS) is a lightweight interpreted programming language with first-class functions.
-
web
Some thing interesting about web. New door for the world.
-
server
A server is a program made to process requests and deliver data to clients.
-
Machine learning
Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.
-
Visualization
Some thing interesting about visualization, use data art
-
Game
Some thing interesting about game, make everyone happy.
Recommend Org
-
Facebook
We are working to build community through open source technology. NB: members must have two-factor auth.
-
Microsoft
Open source projects and samples from Microsoft.
-
Google
Google ❤️ Open Source for everyone.
-
Alibaba
Alibaba Open Source for everyone
-
D3
Data-Driven Documents codes.
-
Tencent
China tencent open source team.
from ultralytics.