Couldn't guess the available gpus on this machine (Windows)

[michael20at]

Hi,

when trying python run.py I get "Couldn't guess the available gpus on this machine".

I use WinPython on Win10. After looking into the utils.py file i see the problem at line 53, the nvidia path for gpu is in linux file system (nvidia_dir = '/proc/driver/nvidia/gpus/').

Any way to get this to work on windows?

The concept sounds great, would love to try it out!

[yburda]

I don't have experience with gpus in windows, but could you call the function with n_gpus=the number of gpus you have?
If you figure out a more general solution, feel free to submit a pull request.

[jinyeom]

I solved this issue by simply adding this in if __name__ == '__main__' of run.py before parser = ...

os.environ['CUDA_VISIBLE_DEVICES'] = '0'

This way, you can control which of your GPUs are visible to your code (e.g., I believe, if you have 2 GPUs, you can replace '0' with '0, 1'). Then, guess_available_gpus will know which GPUs to use.