42 ms·
Does pytorch somehow work better on CUDA? If not, who cares about CUDA?
by sliken 3y ago
Does pytorch somehow work better on CUDA? If not, who cares about CUDA?
- anon291 3y agoYes. to get good perf on pytorch you need to use custom kernels that are written in CUDA, and all the dev work is done in CUDA, so if you want to use new SOTA projects as they come out, you'll probably want an NVIDIA GPU unless you want to spend time hand-optimizing.
- pjmlp 3y agoEveryone else not using Pytorch for their GPU coding activities.