Skip to content

will not run on GPU on COLAB #188

Open
@astromcquinn

Description

@astromcquinn

HI,

I'm having trouble running tensorflow-compression on the COLAB GPUs (which are the only GPUs I have access to). I have the same problem when I run the tensorflow-compression demo script https://www.tensorflow.org/tutorials/generative/data_compression

For example, when I run the provided shell script line to install
read MAJOR MINOR <<< "$(pip show tensorflow | perl -p -0777 -e 's/.Version: (\d+).(\d+)./\1 \2/sg')"
pip install "tensorflow-compression<$MAJOR.$(($MINOR+1))"

and then on the T4 GPU

import tensorflow as tf
import tensorflow_compression as tfc
print(tf.version.VERSION)
print("Num GPUs Available: ", len(tf.config.list_physical_devices('GPU')))

It prints

2.14.1
Num GPUs Available: 0

I don't have this issue in other contexts where I'm not using tensorflow-compression.

Thank you so much,
Matt

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions