From 03829342e283f13ce63e11eb4dc1ffb866d39b65 Mon Sep 17 00:00:00 2001 From: Jade Choghari Date: Thu, 5 Mar 2026 18:37:31 +0300 Subject: [PATCH] add --- .github/workflows/nightly.yml | 2 +- pyproject.toml | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/.github/workflows/nightly.yml b/.github/workflows/nightly.yml index 45bfb9bd5..fc58b0705 100644 --- a/.github/workflows/nightly.yml +++ b/.github/workflows/nightly.yml @@ -188,7 +188,7 @@ jobs: - name: Verify GPU availability run: | nvidia-smi - python -c "import torch; print(f'PyTorch CUDA available: {torch.cuda.is_available()}'); print(f'Number of GPUs: {torch.cuda.device_count()}')" + python -c "import torch; print(f'PyTorch version: {torch.__version__}'); print(f'PyTorch CUDA available: {torch.cuda.is_available()}'); print(f'Number of GPUs: {torch.cuda.device_count()}')" - name: Run multi-GPU training tests # TODO(Steven): Investigate why motors tests are failing in multi-GPU setup diff --git a/pyproject.toml b/pyproject.toml index f4fb7d249..e68998ce9 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -76,7 +76,7 @@ dependencies = [ "pyserial>=3.5,<4.0", "wandb>=0.24.0,<0.25.0", - "torch>=2.2.1,<2.11.0", # TODO: Bump dependency + "torch==2.10.0", "torchcodec>=0.2.1,<0.11.0; sys_platform != 'win32' and (sys_platform != 'linux' or (platform_machine != 'aarch64' and platform_machine != 'arm64' and platform_machine != 'armv7l')) and (sys_platform != 'darwin' or platform_machine != 'x86_64')", # TODO: Bump dependency "torchvision>=0.21.0,<0.26.0", # TODO: Bump dependency