[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
Assignees
Labels
Bug Report
Report bugs detected in Ivy.
Comments
|
@ivy-leaves Please assign me this issue to me so that I can be able to solve this issue. |
|
use ivy.set_default_device(device) |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
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: