(museval-optimization-2) sevagh:sigsep-mus-eval-optimization-2 $ python test.py
size: 4.0 GB
shape: (1000, 1000000)
dtype: float32
OS                           : Linux-5.11.14-100.fc32.x86_64-x86_64-with-glibc2.2.5
CuPy Version                 : 9.0.0
NumPy Version                : 1.20.2
SciPy Version                : 1.6.2
Cython Build Version         : 0.29.22
Cython Runtime Version       : None
CUDA Root                    : /usr
CUDA Build Version           : 11020
CUDA Driver Version          : 11030
CUDA Runtime Version         : 11020
cuBLAS Version               : 11401
cuFFT Version                : 10400
cuRAND Version               : 10203
cuSOLVER Version             : (11, 1, 0)
cuSPARSE Version             : 11400
NVRTC Version                : (11, 2)
Thrust Version               : 101000
CUB Build Version            : 101000
Jitify Build Version         : 60e9e72
cuDNN Build Version          : 8101
cuDNN Version                : 8100
NCCL Build Version           : None
NCCL Runtime Version         : None
cuTENSOR Version             : None
cuSPARSELt Build Version     : None
Device 0 Name                : NVIDIA GeForce RTX 2070 SUPER
Device 0 Compute Capability  : 75
Device 0 PCI Bus ID          : 0000:0A:00.0
None
Traceback (most recent call last):
  File "test.py", line 22, in <module>
    big_array = cupyx.scipy.fft.ifft2(cupy.array(big_array))
  File "/home/sevagh/venvs/museval-optimization-2/lib/python3.8/site-packages/cupyx/scipy/fft/_fft.py", line 214, in ifft2
    return ifftn(x, s, axes, norm, overwrite_x, plan=plan)
  File "/home/sevagh/venvs/museval-optimization-2/lib/python3.8/site-packages/cupyx/scipy/fft/_fft.py", line 285, in ifftn
    return func(x, s, axes, norm, cufft.CUFFT_INVERSE, overwrite_x=overwrite_x,
  File "/home/sevagh/venvs/museval-optimization-2/lib/python3.8/site-packages/cupy/fft/_fft.py", line 585, in _fftn
    a = _convert_dtype(a, value_type)
  File "/home/sevagh/venvs/museval-optimization-2/lib/python3.8/site-packages/cupy/fft/_fft.py", line 37, in _convert_dtype
    a = a.astype(out_dtype)
  File "cupy/_core/core.pyx", line 394, in cupy._core.core.ndarray.astype
  File "cupy/_core/core.pyx", line 453, in cupy._core.core.ndarray.astype
  File "cupy/_core/core.pyx", line 164, in cupy._core.core.ndarray.__init__
  File "cupy/cuda/memory.pyx", line 735, in cupy.cuda.memory.alloc
  File "cupy/cuda/memory.pyx", line 1412, in cupy.cuda.memory.MemoryPool.malloc
  File "cupy/cuda/memory.pyx", line 1433, in cupy.cuda.memory.MemoryPool.malloc
  File "cupy/cuda/memory.pyx", line 1113, in cupy.cuda.memory.SingleDeviceMemoryPool.malloc
  File "cupy/cuda/memory.pyx", line 1134, in cupy.cuda.memory.SingleDeviceMemoryPool._malloc
  File "cupy/cuda/memory.pyx", line 1372, in cupy.cuda.memory.SingleDeviceMemoryPool._try_malloc
cupy.cuda.memory.OutOfMemoryError: Out of memory allocating 8,000,000,000 bytes (allocated so far: 4,000,000,000 bytes).