2 ms·
Given that PyTorch supports AMD cards with ROCm support, I'm curious why would you be coding in CUDA (very low level!) as opposed to PyTorch ?
by HarHarVeryFunny 3y ago
Given that PyTorch supports AMD cards with ROCm support, I'm curious why would you be coding in CUDA (very low level!) as opposed to PyTorch ?
- bri3d 3y ago> a personal non-ML related project Not everything is DNNs and tensors... As for the Stable Diffusion thing, a silly edge case - because MIOpen (and therefore PyTorch-on-ROCm) doesn't work on Windows yet (they're slated to ship it next month, I think).
- HarHarVeryFunny 3y ago> Not everything is DNNs and tensors... Sure, but you could use PyTorch for cuBLAS/cuRAND etc type functionality too.