diff --git a/pystencils/__init__.py b/pystencils/__init__.py index f71103bca09276858a32fe2113f3771121babdb4..82ffaf60a1dcc4d8804ea295766fad84a33092fe 100644 --- a/pystencils/__init__.py +++ b/pystencils/__init__.py @@ -43,10 +43,9 @@ del get_versions # setting the default GPU to the one with maximal memory. GPU_DEVICE is safe to overwrite for different needs try: - import cupy - if cupy.cuda.runtime.getDeviceCount() > 0: - GPU_DEVICE = sorted(range(cupy.cuda.runtime.getDeviceCount()), - key=lambda i: cupy.cuda.Device(i).mem_info[1], reverse=True)[0] + import cupy + if cupy.cuda.runtime.getDeviceCount() > 0: + GPU_DEVICE = sorted(range(cupy.cuda.runtime.getDeviceCount()), + key=lambda i: cupy.cuda.Device(i).mem_info[1], reverse=True)[0] except ImportError: - pass - + pass