2 ms·
With cupy you limit yourself to nvidia hardware. By using pytorch instead you can run on whatever hardware supported by it, either directly or indirectly (e.g.,
by interesaaat 6y ago
With cupy you limit yourself to nvidia hardware. By using pytorch instead you can run on whatever hardware supported by it, either directly or indirectly (e.g., through ONNX conversion)
- fxtentacle 6y agoBut I believe ONNX also doesn't support AMD GPUs, so for people using a regular workstation, the ONNX options are CPU & NVIDIA GPU, just like with numpy and cupy.
- interesaaat 6y agoI am referring to IPU and TPU for example. Or even FPGAs.