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.
2021-09-16 01:07:43 +02:00 · 2021-09-16 01:07:43 +02:00 · 03501a4c8c
parent 9ddab39266
commit 03501a4c8c
2 changed files with 4 additions and 1 deletions
--- 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:
--- 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