"How to Train YOLO-NAS on Custom Dataset" does not build pycocotools

[rafcab]

Search before asking

• I have searched the Roboflow Notebooks issues and found no similar bug report.

Notebook name

How to Train YOLO-NAS on Custom Dataset

Bug

When running the second step in Google Colab, the one titled 'Install YOLO-NAS', gives an error :
ERROR: Could not build wheels for pycocotools, which is required to install pyproject.toml-based projects

Environment

Google Colab

Minimal Reproducible Example

No response

Additional

No response

Are you willing to submit a PR?

• Yes I'd like to help by submitting a PR!

[github-actions]

👋 Hello @rafcab, thank you for leaving an issue on Roboflow Notebooks.

🐞 Bug reports

If you are filing a bug report, please be as detailed as possible. This will help us more easily diagnose and resolve the problem you are facing. To learn more about contributing, check out our Contributing Guidelines.

If you require support with custom code that is not part of Roboflow Notebooks, please reach out on the Roboflow Forum or on the GitHub Discussions page associated with this repository.

💬 Get in touch

Do you have more questions about Roboflow that we haven't responded to yet? Feel free to ask them on the Roboflow Discuss forum. Our developer advocates and community team actively respond to questions there.

To ask questions about Notebooks, head over to the GitHub Discussions section of this repository.

[AhnHyeon-Jin]

I solved this same issue

! pip install git+https://github.com/Deci-AI/super-gradients/releases/tag/3.1.1 --upgrade &> /dev/null
!pip install -q roboflow
!pip install -q supervision

[SkalskiP]

Hi @AhnHyeon-Jin 👋🏻! Awesome! Would you like to create PR with the fix?

[arrafi-musabbir]

I solved this same issue
! pip install git+https://github.com/Deci-AI/super-gradients/releases/tag/3.1.1 --upgrade &> /dev/null

This didn't help in my case on collab. I solved it by following the instructions on their GitHub repo and installed super-gradients like this:
pip install git+https://github.com/Deci-AI/super-gradients.git@stable

[SkalskiP]

Thanks @arrafi-musabbir 🙏🏻 I just updated our YOLO-NAS notebook.