AMD Support?

#40
by Avaredra - opened

Any chance of rocm support for AMD? I tried running the quick install, but the errors resulting indicate nvidia is required.

It works on AMD cards too, but you're basically on your own when it comes to installing the dependencies.

  • Create a virtualenv and activate it
  • Install the correct version of torch and torchaudio from here and then
  • Install the rest of the dependencies with pip install -e .

Sign up or log in to comment