will not run on GPU on COLAB

Open
#188 0 comments 0 reactions 0 assignees View on GitHub

Nobody has claimed this yet.

Assessment

Difficulty
4/5
Estimated time
3-5 days
Newbie friendliness
25/100
Issue type
Bug
Clarity
Needs clarification
Activity status
Stale
Tech stack
python, tensorflow

Research direction

Reproduce the report in Google Colab using the tensorflow-compression demo script and the provided installation command. Check the reported TensorFlow 2.14.1 environment on a T4 GPU and compare GPU visibility with and without tensorflow-compression. Done means the demo environment recognizes the available GPU or the incompatibility is clearly identified.

Written by the indexing model from the issue text.

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

Dominant language
Python
Stars
923
Forks
258
PR merge metrics
No merged PRs in 30d

Contributor guide

Open the contributing guide

First steps

  1. Read the whole issue, then the project's contributing guide.
  2. Comment on the issue to say you are picking it up — it saves two people doing the same work.
  3. Fork the repository and make your change on a branch.
  4. Open a pull request that references the issue number.

More from tensorflow/compression

All issues in tensorflow/compression

Similar issues

More Python issues

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.