Add more pip list

Signed-off-by: Logan Adams <[email protected]>
deepspeedai · Feb 7, 2025 · 6fd2121 · 6fd2121
1 parent b3fa0f5
commit 6fd2121
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/.github/workflows/nv-sd.yml b/.github/workflows/nv-sd.yml
@@ -42,12 +42,16 @@ jobs:
           nvidia-smi
           python -c "import torch; print('torch:', torch.__version__, torch)"
           python -c "import torch; print('CUDA available:', torch.cuda.is_available())"
+          pip list
+          pip install -U numpy==1.26.3
+          pip list
       - name: Install transformers
         run: |
           git clone https://github.com/huggingface/transformers
           cd transformers
           git rev-parse --short HEAD
           python -m pip install .
+          pip list
       - name: Install deepspeed
         run: |
           pip install image-similarity-measures