GPU works with Keras but not Pytorch


(Ghislain) #1

Hello!
When running lesson 1, GPU utilization is at 1%. However, running the keras version makes it go to 50%+.
I followed Jeremy’s howto for Windows.
Also,

  • torch.cuda.isavailable()= True
  • torch.backends.cudnn.enabled= True

OS: Windows 10 Pro
CPU: Ryzen 1920X
GPU: msi 1080 TI

Any help would be very appreciated.

Thanks