Hi team, For your information, pytorch 1.8.0 officially supports ROCm, although it's still in the beta status. https://pytorch.org/ (see install pytorch -> stable -> linux -> pip -> python -> rocm) This looks like an interesting move for the rocm ecosystem.