How to use pytoch for AMD graphics card?

I use amd graphics card, but now I want to learn pytoch. Is there a way to solve this problem?

You need to install the right Pytorch for ROCm. You can find more info on it here:

1 Like