adjust pytorch version number

793eb966 · jnwei · addb80a8 · 793eb966
Commit 793eb966 authored Apr 20, 2024 by jnwei
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

environment.yml environment.yml +1 -1

No files found.
--- a/environment.yml
+++ b/environment.yml
@@ -30,7 +30,7 @@ dependencies:
  - bioconda::hmmer==3.3.2
  - bioconda::hhsuite==3.3.0
  - bioconda::kalign2==2.04
-  - pytorch::pytorch=2.1
+  - pytorch::pytorch=2.2
  - pytorch::pytorch-cuda=12.1
  - pip:
      - deepspeed==0.12.4