search

FHof / torchquad

Multidimensional numerical integration on the GPU using PyTorch
https://www.esa.int/gsp/ACT/open_source/torchquad/
GNU General Public License v3.0
0 stars 0 forks source link

Enable CUDA by default #15

Closed FHof closed 2 years ago

FHof commented 2 years ago

This only affects the torch backend. enable_cuda is still needed in the tests where __init__.py is not executed.

CUDA can be disabled with an environment variable if needed:

CUDA_VISIBLE_DEVICES= python3 -c "from torchquad import enable_cuda; enable_cuda()"