How to use AMD GPU for fastai/pytorch?
Update 3: Since late 2020, torch-mlir project has come a long way and now supports all major Operating systems. Using torch-mlir you can now use your AMD, NVIDIA or Intel GPUs with the latest version of Pytorch. You can download the binaries for your OS from here. Update 2: Since October 21, 2021, You can … Read more