search

openai / jukebox

Code for the paper "Jukebox: A Generative Model for Music"
https://openai.com/blog/jukebox/
Other
7.78k stars 1.4k forks source link

NVIDIA is a requirement? #74

Open josx opened 4 years ago

josx commented 4 years ago

I am running sample.py 

python jukebox/sample.py --model=5b_lyrics --name=sample_5b --levels=3 --sample_length_in_seconds=20 --total_sample_length_in_seconds=180 --sr=44100 --n_samples=6 --hop_fraction=0.5,0.5,0.125

Result:


Using cuda False
Found no NVIDIA driver on your system. Please check that you
have an NVIDIA GPU and installed a driver from
http://www.nvidia.com/Download/index.aspx
``
johndpope commented 4 years ago

pretty much - use google colab / tpu. 12 hrs of gpu compute time = 20 seconds audio. if you have latest imac - still won't help as no macs support cuda.

combs commented 4 years ago

macs running 10.13 or older with nVidia GPUs in the 9xx family do support CUDA, but I don't think it's a version compatible with Jukebox's requirements