Skip to content

Commit c963c21

Browse files
authored
Specify torchvision in nv-ds-chat workflow (prevents errors with torch 2.6) (#6982)
Fixes #6984. The workflow was pulling the updated torch 2.6, which caused CI failures. This keeps us on torch 2.5 for now, since installing torchvision as a dependency later on was pulling torch 2.6 with it which was unintended. This PR also unsets NCCL_DEBUG to avoid a large print out in the case of any errors.
1 parent 065ca8a commit c963c21

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

.github/workflows/nv-ds-chat.yml

+2-1
Original file line numberDiff line numberDiff line change
@@ -37,7 +37,7 @@ jobs:
3737

3838
- name: Install pytorch
3939
run: |
40-
pip3 install -U --cache-dir $TORCH_CACHE torch --index-url https://download.pytorch.org/whl/cu121
40+
pip install -U --cache-dir $TORCH_CACHE torch torchvision --index-url https://download.pytorch.org/whl/cu121
4141
python -c "import torch; print('torch:', torch.__version__, torch)"
4242
python -c "import torch; print('CUDA available:', torch.cuda.is_available())"
4343
@@ -67,6 +67,7 @@ jobs:
6767
run: |
6868
cd DeepSpeedExamples/applications/DeepSpeed-Chat
6969
unset TORCH_CUDA_ARCH_LIST # only jit compile for current arch
70+
unset NCCL_DEBUG
7071
cd tests
7172
pytest $PYTEST_OPTS ./
7273

0 commit comments

Comments
 (0)