From 03501a4c8c5cbe4cd49003af3a77ac0aa1ead5e3 Mon Sep 17 00:00:00 2001 From: henk717 Date: Thu, 16 Sep 2021 01:07:43 +0200 Subject: [PATCH] Dependency Fixes Looks like cudatoolkit is now shipping 11.3, but pytorch has no version for this resulting in an installation of the CPU only version. This is going to lead to people unable to get their GPU running, so for now we force the recommended 11.1 version. I also don't see any harm in allowing Python 3.9 so thats now the default as well to prevent future issues. --- environments/finetuneanon.yml | 3 ++- environments/huggingface.yml | 2 ++ 2 files changed, 4 insertions(+), 1 deletion(-) diff --git a/environments/finetuneanon.yml b/environments/finetuneanon.yml index 7a23b7a2..4e1b4c75 100644 --- a/environments/finetuneanon.yml +++ b/environments/finetuneanon.yml @@ -7,8 +7,9 @@ dependencies: - colorama - flask-socketio - pytorch + - cudatoolkit=11.1 - tensorflow-gpu - - python=3.8.* + - python=3.9.* - pip - git - pip: diff --git a/environments/huggingface.yml b/environments/huggingface.yml index 659ecc74..93398e31 100644 --- a/environments/huggingface.yml +++ b/environments/huggingface.yml @@ -8,6 +8,8 @@ dependencies: - colorama - flask-socketio - pytorch + - python=3.9.* + - cudatoolkit=11.1 - tensorflow-gpu - transformers - pip