Switch to CUDA_VISIBLE_DEVICES instead of PYTOM_GPU

SBC-Utrecht / PyTomPrivate

Development repository

GNU General Public License v2.0

2 stars 2 forks source link

Switch to CUDA_VISIBLE_DEVICES instead of PYTOM_GPU #171

Closed sroet closed 9 months ago

sroet commented 1 year ago

This PR switches to CUDA_VISIBLE_DEVICES instead of PYTOM_GPU like most other softwares.

fixes partly SBC-Utrecht/PyTom#41 (@McHaillet would anything else be needed after this PR?)

McHaillet commented 1 year ago

If the (-g or --gpuID) parameter is not provided by the user, we could make it in the pytom script that the command is then run with CUDA_VISIBLE_DEVICES=-1 (or whatever is used for no available devices). That way we could more easily switch off GPUs for testing.

McHaillet commented 9 months ago

Closing as they should point now to SBC-Utrecht/PyTom