From 6aa7973aecf69a62196a561bb6c3f1f2230b5e00 Mon Sep 17 00:00:00 2001 From: Morgan Funtowicz Date: Mon, 10 Feb 2020 13:24:37 +0100 Subject: [PATCH] Fix circleci cuInit error on Tensorflow >= 2.1.0. Tensorflow 2.1.0 introduce a new dependency model where pip install tensorflow would install tf with GPU support. Before it would just install with CPU support, thus CircleCI is looking for NVidia driver version at initialization of the tensorflow related tests but fails as their is no NVidia Driver running. Signed-off-by: Morgan Funtowicz --- .circleci/config.yml | 6 +++--- setup.py | 1 + 2 files changed, 4 insertions(+), 3 deletions(-) diff --git a/.circleci/config.yml b/.circleci/config.yml index b4d27d2d63..a095d326cb 100644 --- a/.circleci/config.yml +++ b/.circleci/config.yml @@ -10,7 +10,7 @@ jobs: parallelism: 1 steps: - checkout - - run: sudo pip install .[sklearn,tf,torch,testing] + - run: sudo pip install .[sklearn,tf-cpu,torch,testing] - run: sudo pip install codecov pytest-cov - run: python -m pytest -n 8 --dist=loadfile -s -v ./tests/ --cov - run: codecov @@ -26,7 +26,7 @@ jobs: parallelism: 1 steps: - checkout - - run: sudo pip install .[mecab,sklearn,tf,torch,testing] + - run: sudo pip install .[mecab,sklearn,tf-cpu,torch,testing] - run: python -m pytest -n 8 --dist=loadfile -s -v ./tests/ run_tests_torch: working_directory: ~/transformers @@ -52,7 +52,7 @@ jobs: parallelism: 1 steps: - checkout - - run: sudo pip install .[sklearn,tf,testing] + - run: sudo pip install .[sklearn,tf-cpu,testing] - run: sudo pip install codecov pytest-cov - run: python -m pytest -n 8 --dist=loadfile -s -v ./tests/ --cov - run: codecov diff --git a/setup.py b/setup.py index b36d51e719..81fb09f852 100644 --- a/setup.py +++ b/setup.py @@ -63,6 +63,7 @@ extras = {} extras["mecab"] = ["mecab-python3"] extras["sklearn"] = ["scikit-learn"] extras["tf"] = ["tensorflow"] +extras["tf-cpu"] = ["tensorflow-cpu"] extras["torch"] = ["torch"] extras["serving"] = ["pydantic", "uvicorn", "fastapi", "starlette"]