search

msieg / deep-music-visualizer

The Deep Visualizer uses BigGAN (Brock et al., 2018) to visualize music.
384 stars 104 forks source link

Running on GPU #29

Open Gitler101 opened 1 year ago

Gitler101 commented 1 year ago

I got this running and enjoy it, but I would like to run it on my GPU. Is this possible? I played around a bit with numba which is supposed to make it possible but I am not sure of which changes I would have to make in order to get this to work.

CameronSima commented 1 year ago

I believe it would depend on what hardware you're using. I'm on a MacBook Pro M1 and I changed this line:


#set device
device = torch.device('mps')
#device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')

I think cuda refers to NVIDIA gpus