Merge branch 'ko3n1g/fix/setup' into 'main'

fix: Setup requirements See merge request ADLR/megatron-lm!2579
NVIDIA · Jan 22, 2025 · 5c12382 · 5c12382
2 parents 27756f4 + 4df1108
commit 5c12382
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/setup.py b/setup.py
@@ -29,7 +29,7 @@
 
 
 def req_file(filename, folder="requirements"):
-    environment = os.getenv("PY_ENV", "pytorch:24.07")
+    environment = os.getenv("PY_ENV", "pytorch_24.10")
 
     content = []
     with open(os.path.join(folder, environment, filename), encoding='utf-8') as f: