diff --git a/environment.yml b/environment.yml index f8080ae..f040cf4 100644 --- a/environment.yml +++ b/environment.yml @@ -12,7 +12,6 @@ dependencies: # Extra requirements - cudatoolkit=10.2 - - tensorflow-gpu - pytorch - torchvision - keras @@ -28,7 +27,6 @@ dependencies: - wget - ramp-utils - imbalanced-learn - - tensorflow-addons - bids - osfclient # Extra requirements pip diff --git a/extra_libraries.txt b/extra_libraries.txt index 956c6b8..74ea467 100644 --- a/extra_libraries.txt +++ b/extra_libraries.txt @@ -5,6 +5,4 @@ keras lightgbm xgboost catboost -tensorflow-gpu -tensorflow-addons torchvision