diff --git a/setup.py b/setup.py index e795f5b93d..d4d7cb3346 100644 --- 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: