ledatelescope / bifrost

A stream processing framework for high-throughput applications.
BSD 3-Clause "New" or "Revised" License
66 stars 29 forks source link

gpu allocation #148

Open telegraphic opened 3 years ago

telegraphic commented 3 years ago

I couldn’t get bifrost code to fully execute on a GPU other than GPU=0, but if you set an environment variable like this before running the code, it will work correctly:

CUDA_VISIBLE_DEVICES=2 spip_correlator.py

(Not sure if this is just our setup)