-
Notifications
You must be signed in to change notification settings - Fork 621
Open
Description
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!
Metadata
Metadata
Assignees
Labels
No labels