4 ms·

How about Pytorch?

• jamesblonde 7y ago
  You have to compile it yourself, which sucks. For some reason PyTorch aren't agreeing to upstream ROCm, yet.
  • bgorman 7y ago
    You don't need to compile it, AMD provides docker images. For example I was able to run fast.ai/pytorch oh my amd gpu https://github.com/briangorman/fastai_rocm_docker https://github.com/briangorman/fastai_rocm_docker