Description
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