apple / apple/corenet

Corenet detect M2 GPU

Open
#26 1 comment 0 reactions 0 assignees View on GitHub
Dominant language
Jupyter Notebook
Stars
7k
Forks
541
PR merge metrics
No merged PRs in 30d

Description

I run the "train_a_new_model_on_a_new_dataset_from_scratch" notebook on my M2 and no GPUs are detected, the script moves to using CPUs. Is it possible to configure Corenet to detect the M2 GPU?

Contributor guide

Open the contributing guide

Research direction

Start with the train_a_new_model_on_a_new_dataset_from_scratch notebook and reproduce the issue on an M2 system. Trace how Corenet checks for available GPUs and document or implement the configuration needed for the notebook to select the M2 GPU instead of CPUs; done means training reports and uses the GPU.

Written by the indexing model from the issue text.

Assessment

Tech stack
jupyter-notebook
Domain
machine-learning
Issue type
Bug
Difficulty
4/5
Estimated time
3-5 days
Activity status
Stale
Clarity
Needs clarification
Newbie friendliness
35/100

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.