RuntimeError: No CUDA GPUs are available

[arminbiglari]

Dear author, Thank you for your great work.

Since i want to consider this work as one of my baselines, during running the code, i faced this error : RuntimeError: No CUDA GPUs are available.

i managed to solved the issue, which might be useful for others.

to overcome this issue while training using a single gpu, please add os.environ["CUDA_VISIBLE_DEVICES"] = "0" to line 508 in main.py

best wishes