update

update

update

update

update

update

update

Author: akihironitta

.github/actions/setup/action.yml

@@ -28,20 +28,27 @@ runs:
       if: ${{ inputs.torch-version != 'nightly' }}
       run: |
         pip install torch==${{ inputs.torch-version }}.* --extra-index-url https://download.pytorch.org/whl/${{ inputs.cuda-version }}
-        python -c "import torch; print('PyTorch:', torch.__version__)"
-        python -c "import torch; print('CUDA:', torch.version.cuda)"
       shell: bash
 
     - name: Install PyTorch ${{ inputs.torch-version }}+${{ inputs.cuda-version }}
       if: ${{ inputs.torch-version == 'nightly' }}
       run: |
         pip install --pre torch --extra-index-url https://download.pytorch.org/whl/nightly/${{ inputs.cuda-version }}
-        python -c "import torch; print('PyTorch:', torch.__version__)"
-        python -c "import torch; print('CUDA:', torch.version.cuda)"
+      shell: bash
+
+    - name: Downgrade NumPy if necessary
+      run: |
+        torch_version=$(python -c "import torch; print(torch.__version__)")
+        if [[ $torch_version == *"2.0"* || $torch_version == *"2.1"* || $torch_version == *"2.2"* ]]; then
+          pip install 'numpy<2'
+        fi
       shell: bash
 
     - name: List installed packages
-      run: pip list
+      run: |
+        pip list
+        python -c "import torch; print('PyTorch:', torch.__version__)"
+        python -c "import torch; print('CUDA:', torch.version.cuda)"
       shell: bash
 
     # TODO: Include catboost in Python 3.13 CI when catboost supports it: