Merge pull request #50 from graphcore-research/freeze-torch-version

Fix torch version at 2.1 for now
graphcore-research · Apr 16, 2024 · dfbbe7a · dfbbe7a
2 parents e77bb5d + ab67119
commit dfbbe7a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/requirements.txt b/requirements.txt
@@ -4,4 +4,4 @@ einops
 numpy
 seaborn
 tabulate
-torch
+torch==2.1