search

ivy-llc / ivy

Convert Machine Learning Code Between Frameworks
https://ivy.dev
Other
14.02k stars 5.77k forks source link

[Bug]: Device is set to gpu:0 but during training model is not using the GPU on goolge colab #28610

Open marvlyngkhoi opened 6 months ago

marvlyngkhoi commented 6 months ago

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

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

Kamalesh3112 commented 6 months ago

@ivy-leaves Please assign me this issue to me so that I can be able to solve this issue.

abhi2596 commented 6 months ago

use ivy.set_default_device(device)