4 ms·
> Isn't there a way to dynamically select between numpy and cupy, depending on whether you want cpu or gpu code? CuPy is an (almost) drop-in replacement for Nu
by johndough 2y ago
> Isn't there a way to dynamically select between numpy and cupy, depending on whether you want cpu or gpu code?
CuPy is an (almost) drop-in replacement for NumPy, so the following works surprisingly often:
if use_cpu:
import numpy as np
else:
import cupy as np
- gnulinux 2y ago> surprisingly This is the problem with these kind of methods. It works, until it doesn't in an unknown way.