Programming thread

[Anti Snigger]

I’m using a CNN to train autonomous drones on a course using RGB-D input in stable-baselines3, I’d like to know if I can use my Mac for GPU training or if I need to use a dedicated graphics card.

what GPU do you have, and what kind of dataset are you using, alongside network architecture?

 

[Creative Username]

I’d like to know if I can use my Mac for GPU training or if I need to use a dedicated graphics card.

i did a cursory search and i think you just have to find a version of pytorch with support for the graphics api that comes in your mac (which depends a lot on how old it is, and the specific gpu)
then in your program you have to run a "initialize with gpu" function as opposed to the regular "initialize with cpu" function

 

[canalCrab]

i did a cursory search and i think you just have to find a version of pytorch with support for the graphics api that comes in your mac (which depends a lot on how old it is, and the specific gpu)
then in your program you have to run a "initialize with gpu" function as opposed to the regular "initialize with cpu" function

Only slightly related: does pytorch even have a m3 version?

 

[Creative Username]

Only slightly related: does pytorch even have a m3 version?

i saw something about a metal version while i was doing my cursory search

 

[Fcret]

Only slightly related: does pytorch even have a m3 version?

Yes - https://developer.apple.com/metal/pytorch/