Hi,
Thank you for the awesome work! It is very helpful!
Is it possible to run the code on GPU? I am running it on Ubuntu 18.04 and when I try to run, it does not run on gpu devices and runs on cpu. Is there any way to enforce the condition to run the example.py on gpu? I also tried to run export CUDA_VISIBLE_DEVICES=0, but it still seems to be running on cpu.
Many thanks!
Hi,
Thank you for the awesome work! It is very helpful!
Is it possible to run the code on GPU? I am running it on Ubuntu 18.04 and when I try to run, it does not run on gpu devices and runs on cpu. Is there any way to enforce the condition to run the example.py on gpu? I also tried to run export CUDA_VISIBLE_DEVICES=0, but it still seems to be running on cpu.
Many thanks!