You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
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
The text was updated successfully, but these errors were encountered:
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
I don't have this issue in other contexts where I'm not using tensorflow-compression.
Thank you so much,
Matt
The text was updated successfully, but these errors were encountered: