will not run on GPU on COLAB #188

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