Issue sampling my own music into OpenAI's Jukebox using Anaconda prompt

Question

I'm having issues importing my own samples into Jukebox (https://github.com/openai/jukebox/) and received this error message when attempting to select my files. "AttributeError: module 'torch.distributed' has no attribute 'ReduceOp'"

Here is a screenshot: https://www.dropbox.com/s/o0cdlpzo3n0u4yg/jukeboxcodeproblem.png?dl=0

Anyone know how to fix this? I've followed all the steps in the readme.md file (https://github.com/openai/jukebox/) up until this point. I'm very new to coding so all help is very appreciated.

Answer 1

Looks like I have the same issue. From what I got, pytorch.distributed is not supported on Windows: https://pytorch.org/docs/stable/distributed.html

If you try smth like:

> python
>>> import torch
>>> import torch.ditributed as dist
>>> dist.is_available()

You'll get "False"

A bit more info can be found here: https://github.com/NVIDIA/apex/issues/429

So, to make it work you need Linux or somehow get rid of torch distributed.