Comment by leopoldj

The blog doesn't verify if the code is actually using the GPU. The code will work perfectly fine on CPU, albeit slowly. You should run this to be sure:

    python -c "import torch; print(torch.cuda.is_available())"

Strange that torch.cuda.is_available() is used for AMD also.

Use rocm-smi to be double sure.