Comment Comment by postmeta pytorch already supports AMD with device=cudaalready opensourced ROCm/HIPhttps://github.com/pytorch/pytorch/blob/fb8876069d89aaf27cc9... parent postmeta 2 years ago
already opensourced ROCm/HIP
https://github.com/pytorch/pytorch/blob/fb8876069d89aaf27cc9...