40 ms·
You have to compile it yourself, which sucks. For some reason PyTorch aren't agreeing to upstream ROCm, yet.
by jamesblonde 7y ago
You have to compile it yourself, which sucks. For some reason PyTorch aren't agreeing to upstream ROCm, yet.
- bgorman 7y agoYou 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