Comment by nl
3 hours ago
I suspect the OP's issues might be mostly related to the ROCM version of PyTorch. AMD still can't get this right.
3 hours ago
I suspect the OP's issues might be mostly related to the ROCM version of PyTorch. AMD still can't get this right.
Probably - but the answer is to avoid ROCM, not pytorch.