[Bug]: Device is set to gpu:0 but during training model is not using the GPU on goolge colab #28610
Open
1 of 4 tasks
Labels
Bug Report
Report bugs detected in Ivy.
Bug Explanation
I'm trying to create demos using ivy framework to build a CNN Model for plant disease detection. Everything is Working fine on CPU as intended except that when device is set to gpu:0 model is not using GPU and its still training on CPU as shown below
Steps to Reproduce Bug
Here is the colab notebook https://colab.research.google.com/drive/1IY5L0ZUV7HDb2Cv6dt1B0lwb2uDMKMqt?usp=sharing to easily reproduce the bug
Environment
Colab Notebook
Ivy Version
0.0.8.0
Backend
Device
No response
The text was updated successfully, but these errors were encountered: