From a52f392ce7bca8deb5698f2fad1577af73619b89 Mon Sep 17 00:00:00 2001 From: "Ta, Kevin" Date: Fri, 7 Jul 2023 14:53:44 -0700 Subject: [PATCH] removed a leftover environment variable call, it should only be set in the job command --- .../acpt-pytorch-2.0-cuda11.7/tests/pytorch2_0_sample_test.py | 1 - 1 file changed, 1 deletion(-) diff --git a/assets/training/general/environments/acpt-pytorch-2.0-cuda11.7/tests/pytorch2_0_sample_test.py b/assets/training/general/environments/acpt-pytorch-2.0-cuda11.7/tests/pytorch2_0_sample_test.py index 47b95cd252..03d82163f7 100644 --- a/assets/training/general/environments/acpt-pytorch-2.0-cuda11.7/tests/pytorch2_0_sample_test.py +++ b/assets/training/general/environments/acpt-pytorch-2.0-cuda11.7/tests/pytorch2_0_sample_test.py @@ -14,7 +14,6 @@ JOB_SOURCE_CODE = "../../acpt-tests/src" TIMEOUT_MINUTES = os.environ.get("timeout_minutes", 60) STD_LOG = Path("artifacts/user_logs/std_log.txt") -os.environ['DS_ACCELERATOR'] = 'cuda' def test_pytorch_2_0():