search

vanvalenlab / deepcell-tf

Deep Learning Library for Single Cell Analysis
https://deepcell.readthedocs.io
Other
413 stars 95 forks source link

Notebooks don't connect with Google Colab's GPU #383

Closed esgomezm closed 3 years ago

esgomezm commented 4 years ago

Is your feature request related to a problem? Please describe. Whenever I run the notebooks using Google Colab, in the training I get the following message: 

X_train shape: (90, 216, 256, 1)
y_train shape: (90, 216, 256, 1)
X_test shape: (10, 216, 256, 1)
y_test shape: (10, 216, 256, 1)
Output Shape: (None, 216, 256, 2)
Number of Classes: 2
Training on 0 GPUs

and Google Colabs sends a warning message about being on a GPU environment without using any GPU.

Describe the solution you'd like It would be nice if the code connects with their GPUs/TPUs.

willgraf commented 3 years ago

Following up on this, I think it's likely that tensorflow was installed rather than tensorflow-gpu. The two packages have been merged with TensorFlow 2, and this issue should be resolved.

The next release of deepcell (based on TensorFlow 2.3.1) is now a pip package to make Colab installation easy.

willgraf commented 3 years ago

This is now resolved in release 0.8.2.