4 ms·

I suspect the OP's issues might be mostly related to the ROCM version of PyTorch. AMD still can't get this right.

• danielmarkbruce 1y ago
  Probably - but the answer is to avoid ROCM, not pytorch.
  • yorwba 1y ago
    Avoiding ROCm means buying a new Nvidia GPU. Some people would like to keep using the hardware they already have.
    • danielmarkbruce 1y ago
      The cost to deal with rocm is > cost of a consumer nvidia gpu by orders of magnitude.