Open Vignesh048 opened 1 year ago
do you have any GPU?
@Vignesh048 : You may change the os.environ['CUDA_VISIBLE_DEVICES'] = "3" to os.environ['CUDA_VISIBLE_DEVICES'] = "0" and verify the execution. It should work provided you have installed the pytorch in GPU mode.
I'm facing the following issue while running the code for training.