Skip to content
Snippets Groups Projects
Commit 65fb3141 authored by Alexander Gehrke's avatar Alexander Gehrke
Browse files

Remove extras_requires by using TF > 2.0

As the code runs with any tensorflow version from 2.0 to 2.5, the
tensorflow-gpu package is only needed if it isn't possible to use
TF>=2.1 for some external reason. In that case, it's up to the user to
install the gpu version, as this simplifies installation for all other
cases.
parent e6067744
No related branches found
No related tags found
No related merge requests found
...@@ -4,10 +4,11 @@ ...@@ -4,10 +4,11 @@
Python dependencies are specified in `requirements.txt` / `setup.py`. Python dependencies are specified in `requirements.txt` / `setup.py`.
You must install the package via pip with either `ocr4all_pixel_classifier[tf_cpu]` to The package is tested with Tensorflow 2.0 up to 2.5. If you want to use a GPU,
use CPU version of tensorflow or `ocr4all_pixel_classifier[tf_gpu]` to use GPU (CUDA) you'll have to set up your system with the CUDA and CuDNN versions [matching
version of tensorflow. For the latter, your system should be set up with CUDA 9 your used Tensorflow version](https://www.tensorflow.org/install/source#gpu).
and CuDNN 7. If using Tensorflow older than 2.1 for some reason, you'll additionaly have to
replace the `tensorflow` package with `tensorflow-gpu` manually.
## Usage ## Usage
......
...@@ -8,4 +8,4 @@ dataclasses ...@@ -8,4 +8,4 @@ dataclasses
efficientnet efficientnet
lxml lxml
ocr4all-pylib>=0.2.0 ocr4all-pylib>=0.2.0
tensorflow>=2.0.0 tensorflow >= 2.0.0, <= 2.5.0
...@@ -14,11 +14,7 @@ setup( ...@@ -14,11 +14,7 @@ setup(
author="Christoph Wick, Alexander Hartelt, Alexander Gehrke", author="Christoph Wick, Alexander Hartelt, Alexander Gehrke",
author_email="christoph.wick@informatik.uni-wuerzburg.de, alexander.hartelt@informatik.uni-wuerzburg.de, alexander.gehrke@informatik.uni-wuerzburg.de", author_email="christoph.wick@informatik.uni-wuerzburg.de, alexander.hartelt@informatik.uni-wuerzburg.de, alexander.gehrke@informatik.uni-wuerzburg.de",
url="https://gitlab2.informatik.uni-wuerzburg.de/chw71yx/page-segmentation.git", url="https://gitlab2.informatik.uni-wuerzburg.de/chw71yx/page-segmentation.git",
install_requires=open("requirements.in").read().split(), install_requires=open("requirements.in").read().split("\n"),
extras_require={
'tf_cpu': ['tensorflow>=2.0.0,<2.1.0'],
'tf_gpu': ['tensorflow-gpu>=2.0.0,<2.1.0'],
},
classifiers=[ classifiers=[
"Programming Language :: Python :: 3", "Programming Language :: Python :: 3",
"License :: OSI Approved :: GNU General Public License v3 or later (GPLv3+)", "License :: OSI Approved :: GNU General Public License v3 or later (GPLv3+)",
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment