[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
![Screenshot from 2024-03-15 14-14-29](https://private-user-images.githubusercontent.com/78148524/313124624-b74e5d68-f689-48a5-b5bd-00761d1d6af1.png?jwt=eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJpc3MiOiJnaXRodWIuY29tIiwiYXVkIjoicmF3LmdpdGh1YnVzZXJjb250ZW50LmNvbSIsImtleSI6ImtleTUiLCJleHAiOjE3MjAzNDA5NzYsIm5iZiI6MTcyMDM0MDY3NiwicGF0aCI6Ii83ODE0ODUyNC8zMTMxMjQ2MjQtYjc0ZTVkNjgtZjY4OS00OGE1LWI1YmQtMDA3NjFkMWQ2YWYxLnBuZz9YLUFtei1BbGdvcml0aG09QVdTNC1ITUFDLVNIQTI1NiZYLUFtei1DcmVkZW50aWFsPUFLSUFWQ09EWUxTQTUzUFFLNFpBJTJGMjAyNDA3MDclMkZ1cy1lYXN0LTElMkZzMyUyRmF3czRfcmVxdWVzdCZYLUFtei1EYXRlPTIwMjQwNzA3VDA4MjQzNlomWC1BbXotRXhwaXJlcz0zMDAmWC1BbXotU2lnbmF0dXJlPTgxMmM5ZmFjZWJkMjUzYTUzYzk2OTk1M2FkYjc4MTJmNjEwNWNjN2I2ODUwMjhhYTYyMjFlZjcyZWJjZDE1ZWEmWC1BbXotU2lnbmVkSGVhZGVycz1ob3N0JmFjdG9yX2lkPTAma2V5X2lkPTAmcmVwb19pZD0wIn0.az2biXAsXGZ8UOBT_ibQj41rgNKJ0RKN2KuGc1E8PIk)
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: