debug installing torch for Thunder (#2072)

Author: Borda

.azure/gpu-test.yml

@@ -63,13 +63,13 @@ jobs:
 
       - script: |
           pip install --upgrade pip
-          pip install '.[extra,test]' cffi -U
+          pip install '.[extra,test]' "torch==${TORCH_VERSION}" cffi -U
         displayName: "Install package & dependencies"
 
       - script: |
           set -e
           pip uninstall -y torchvision torchaudio
-          pip install '.[compiler]'
+          pip install '.[compiler,extra,test]' "torch==${TORCH_VERSION}"
           python -c "from thunder.executors import nvfuser_available ; assert nvfuser_available(), 'nvFuser is missing!'"
           python -c "from thunder.executors.triton_utils import triton_version ; assert triton_version() is not None, 'triton is missing!'"
         condition: eq(variables['dependency'], 'compiler')
@@ -98,7 +98,7 @@ jobs:
       - bash: |
           pip uninstall -y lightning-thunder
           # install thunder from source, so that, thunder.tests will be available
-          pip install -U "lightning-thunder[test] @ git+https://github.com/Lightning-AI/lightning-thunder.git"
+          pip install -U "lightning-thunder[test] @ git+https://github.com/Lightning-AI/lightning-thunder.git" "torch==${TORCH_VERSION}"
         displayName: "Re-install Thunder [main branch]"
         condition: eq(variables['dependency'], 'compiler')