How to use AMD GPU for fastai/pytorch?

pytorch gpu amd fast-ai
Mohanned ElSayed picture Mohanned ElSayed · Jul 21, 2020 · Viewed 19.7k times · Source

I'm using a laptop which has Intel Corporation HD Graphics 5500 (rev 09), and AMD Radeon r5 m255 graphics card.

Does anyone know how to it set up for Deep Learning, specifically fastai/Pytorch?

Answer

Rika picture Rika · Jul 21, 2020

Update :

As of Pytorch 1.8 (March 04, 2021), AMD ROCm versions are made available from Pytorch's official website. You can now easily install them on Linux and Mac, the same way you used to install the CUDA/CPU versions.

Currently, the pip packages are being provided only. Also, the Mac and Windows platforms are still not supported (I haven't tested with WSL2 though!)

Old answer:

You need to install the ROCm version. The official AMD instructions on building Pytorch is here.

There was previously a wheel package for rocm, but it seems AMD doesn't distribute that anymore, and instead, you need to build PyTorch from the source as the guide which I linked to explains.

However, you may consult this page, to build the latest PyTorch version: The unofficial page of ROCm/PyTorch.

Related questions

How to check if pytorch is using the GPU?

I would like to know if pytorch is using my GPU. It's possible to detect with nvidia-smi if there is any activity from the GPU during the process, but I want something written in a python script. Is there a …

Read More
CUDA runtime error (59) : device-side assert triggered

I have access to Tesla K20c, I am running ResNet50 on CIFAR10 dataset... Then I get the error as: THCudaCheck FAIL file=/opt/conda/conda-bld/pytorch_1524584710464/work/aten/src/THC/generated/../generic/THCTensorMathPointwise.cu line=265 error=59 : device-side assert triggered …

Read More
pytorch delete model from gpu

I want to make a cross validation in my project based on Pytorch. And I didn't find any method that pytorch provided to delete the current model and empty the memory of GPU. Could you tell that how can I …

Read More