ivy icon indicating copy to clipboard operation
ivy copied to clipboard

Published 20 hours ago • verified publisher icon Transpile-AI

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

Open marvlyngkhoi opened this issue 11 months ago • 2 comments

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

  • [ ] NumPy
  • [X] TensorFlow
  • [ ] PyTorch
  • [ ] JAX

Device

No response

marvlyngkhoi avatar Mar 15 '24 08:03 marvlyngkhoi

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

Kamalesh3112 avatar Apr 08 '24 06:04 Kamalesh3112

use ivy.set_default_device(device)

abhi2596 avatar Apr 09 '24 14:04 abhi2596