Проблемы с зависимостями TensorFlow при использовании Colab

Question

Я пытаюсь использовать модель пакетного ансамбля с Colab. Установка пипса:

      !pip install "git+https://github.com/google/uncertainty-baselines.git#egg=uncertainty_baselines"

и получил следующую ошибку:

      ERROR: tensorflow 2.5.0 has requirement keras-nightly~=2.5.0.dev, but you'll have keras-nightly 2.7.0.dev2021070800 which is incompatible.
ERROR: tf-nightly 2.7.0.dev20210708 has requirement grpcio<2.0,>=1.37.0, but you'll have grpcio 1.34.1 which is incompatible.

Как только я пытаюсь использовать библиотеку с помощью импорта:

      import uncertainty_baselines as ub

это вызывает эту ошибку

      ImportError: cannot import name '__version__' from 'keras' (/usr/local/lib/python3.7/dist-packages/keras/__init__.py)

Поэтому я попытался вручную исправить версии пакетов, я попытался сначала установить следующие пакеты и зависимости:

      !pip install keras-nightly==2.5.0.dev2021020510
!pip install grpcio==1.34.0
!pip install tensorflow==2.5.0
!pip install tf-nightly==2.7.0.dev20210708

но это вызывает следующие ошибки:

      ERROR: tf-nightly 2.7.0.dev20210708 has requirement grpcio<2.0,>=1.37.0, but you'll have grpcio 1.34.1 which is incompatible.
ERROR: tf-nightly 2.7.0.dev20210708 has requirement keras-nightly~=2.7.0.dev, but you'll have keras-nightly 2.5.0.dev2021020510 which is incompatible

Опять же попробовал исправить версии:

      !pip install keras-nightly==2.7.0.dev2021070800
!pip install grpcio==1.37.0
!pip install tensorflow==2.5.0
!pip install tf-nightly==2.7.0.dev20210708

Но теперь возникают следующие ошибки:

      ERROR: tensorflow 2.5.0 has requirement grpcio~=1.34.0, but you'll have grpcio 1.37.0 which is incompatible.
ERROR: tensorflow 2.5.0 has requirement keras-nightly~=2.5.0.dev, but you'll have keras-nightly 2.7.0.dev2021070800 which is incompatible.

Это похоже на круговые зависимости, которые я не могу решить и связанные с тензором, а не с конкретным репозиторием, который я пытаюсь использовать.

Answer 1

Проблема была решена с новым MR -github.com/google/uncerty-baselines/issues/407

Зависимость tf-nightly была удалена.