Train Pytorch with GPU on Apple Silicon (M1 series)

preview_player
Показать описание
Finally, pytorch team has announced support for Apple Silicon GPU support. See how much speed gain you can get with your m1 computer.

Announcement

Installation

Pytorch examples - MNIST
Рекомендации по теме
Комментарии
Автор

Is it relevant with to fix ERROR: to 0.0 on my MacBook Pro M3 Pro?

성이름-gmd
Автор

I tried this before of this pytorch support with M1 and tensorflow metal or another Apple "suggestions" and it didn't work, and I had a lot of problems but I congratulate you if you can now use that processor for GPU tasks but I think it doesn't compare with the training with a minimal GPU capability in a free Google Colab server or with a Gaming PC

johnorozco
Автор

I tried it on ultralytics yolov5 and it doesn't work yet. they're making progress and hopefully by end of the year it will work on yolov5

woolfel
Автор

I ran the command --device mpa, I got below error.
RuntimeError: PyTorch is not linked with support for mps devices
torch_version: '1.12.0.dev20220520'
Installed torch using anaconda

mohammedsohilshaikh
Автор

Ok, seems rather hopeless for my apple mac air. Anyway, for the price, sort of expected...

tildarusso
Автор

Hi, nice video, can you try with a different or bigger batch size...

sabirdvd
Автор

My torch version is 1.12.0.dev20220518 but cuda still using CPU

michaelshi