From 5efec638fe7698f36ced04c1300212f204f8eaed Mon Sep 17 00:00:00 2001 From: Mark Kurtz Date: Wed, 3 Feb 2021 16:11:11 -0500 Subject: [PATCH] Update docs for tensorflow v1 issues --- Makefile | 2 +- README.md | 2 ++ 2 files changed, 3 insertions(+), 1 deletion(-) diff --git a/Makefile b/Makefile index c2afac514ec..9f471852123 100644 --- a/Makefile +++ b/Makefile @@ -7,7 +7,7 @@ DOCDIR := docs MDCHECKGLOBS := 'docs/**/*.md' 'docs/**/*.rst' 'examples/**/*.md' 'notebooks/**/*.md' 'scripts/**/*.md' MDCHECKFILES := CODE_OF_CONDUCT.md CONTRIBUTING.md DEVELOPING.md README.md -TARGETS := "" # targets for running pytests: keras,onnx,pytorch,pytorch_models,pytorch_datasets,tensorflow_v1,tensorflow_v1_models,tensorflow_v1_datasets +TARGETS := "" # targets for running pytests: keras,onnx,pytorch,pytorch_models,pytorch_datasets,tensorflow_v1,tensorflow_v1_datasets PYTEST_ARGS := "" ifneq ($(findstring keras,$(TARGETS)),keras) PYTEST_ARGS := $(PYTEST_ARGS) --ignore tests/sparseml/keras diff --git a/README.md b/README.md index 75e6e096995..1fa10bf2177 100644 --- a/README.md +++ b/README.md @@ -331,6 +331,8 @@ pip install sparseml[tf_v1_gpu] Depending on your device and CUDA version, you may need to install additional dependencies for using TensorFlow V1 with GPU operations. You can find these steps [here](https://www.tensorflow.org/install/gpu#older_versions_of_tensorflow). +Note, TensorFlow V1 is no longer being built for newer operating systems such as Ubuntu 20.04. Therefore, SparseML with TensorFlow V1 is unsupported on these operating systems as well. + ## Resources and Learning More - [SparseZoo Documentation](https://docs.neuralmagic.com/sparsezoo/)