# Unable to start federated training on CUDA11.8 when using YOLOv8 model on NVIDIA A10G GPU

**URL:** <https://discuss.flower.ai/t/unable-to-start-federated-training-on-cuda11-8-when-using-yolov8-model-on-nvidia-a10g-gpu/133>\
**Category:** Flower Help - Beginners\
**Created:** [March 12, 2024, 8:52am UTC](https://discuss.flower.ai/t/unable-to-start-federated-training-on-cuda11-8-when-using-yolov8-model-on-nvidia-a10g-gpu/133 "2024-03-12T08:52:31Z")\
**Posts on this page:** 3\
**Page:** 1

<div class="post-metadata">

**Author:** ![shubham.ecebtech14](https://avatars.discourse-cdn.com/v4/letter/s/a4c791/32.png) [@shubham.ecebtech14](https://discuss.flower.ai/u/shubham.ecebtech14)\
**Post date:** [March 12, 2024, 8:52am UTC](https://discuss.flower.ai/t/unable-to-start-federated-training-on-cuda11-8-when-using-yolov8-model-on-nvidia-a10g-gpu/133/1 "2024-03-12T08:52:31Z")

</div>

I wish to report one strange observation. I was able to run my federated code with available CPU. Today, I installed CUDA 11.8 and tried to run the code. It is getting stuck and not able to start training.  
To reproduce the issue, I again tried running with CPU only and it was working fine.  
I have installed cuda and cudaCNN compatible with pytorch after checking on official pytorch website and they were installed successfully without any warning or error message. I am using an AWS instance windows XP 2022 server with NVIDIA A10G gpu. When i run any other centralized learning code on cuda, it works fine. But when i run federated learning code, it gives error. The training does not start and then grpc channel closes after waiting for sometime.

---

<div class="post-metadata">

**Author:** ![javier](https://dub1.discourse-cdn.com/flex013/user_avatar/discuss.flower.ai/javier/32/116_2.png) [@javier](https://discuss.flower.ai/u/javier)\
**Post date:** [March 20, 2024, 8:33pm UTC](https://discuss.flower.ai/t/unable-to-start-federated-training-on-cuda11-8-when-using-yolov8-model-on-nvidia-a10g-gpu/133/2 "2024-03-20T20:33:11Z")

</div>

Hi @shubham.ecebtech14, could you paste here the error you obtain ? What version of flower are you using ? and what version of Python? Have you tried running the [`examples/simulation-pytorch`](https://github.com/adap/flower/tree/main/examples/simulation-pytorch) ?

---

<div class="post-metadata">

**Author:** ![williamlm](https://dub1.discourse-cdn.com/flex013/user_avatar/discuss.flower.ai/williamlm/32/162_2.png) [@williamlm](https://discuss.flower.ai/u/williamlm)\
**Post date:** [August 1, 2025, 3:29pm UTC](https://discuss.flower.ai/t/unable-to-start-federated-training-on-cuda11-8-when-using-yolov8-model-on-nvidia-a10g-gpu/133/3 "2025-08-01T15:29:30Z")

</div>

Hi @shubham.ecebtech14,

is this an error that you’re still experiencing? If so, could you paste the output log here? Feel free to also report this on GitHub Issues.

Best regards,

William