up

microsoft · Jun 12, 2024 · a86a3e6 · a86a3e6
1 parent 94aa21c
commit a86a3e6
Showing 1 changed file with 8 additions and 8 deletions.
diff --git a/tools/ci_build/github/azure-pipelines/templates/jobs/download_win_gpu_library.yml b/tools/ci_build/github/azure-pipelines/templates/jobs/download_win_gpu_library.yml
@@ -35,19 +35,19 @@ steps:
 
   - ${{ if eq(parameters.DownloadTRT, true) }}:
     - ${{ if eq(parameters.CudaVersion, '11.8') }}:
-        - bash: |
-            echo "##vso[task.setvariable variable=trtCudaVersion]11.8"
+        - powershell: |
+            Write-Host "##vso[task.setvariable variable=trtCudaVersion;]11.8"
           displayName: Set trtCudaVersion
     - ${{ if and(eq(parameters.CudaVersion, '12.2'), eq(parameters.TrtVersion, '8.6.1.6')) }}:
-        - bash: |
-            echo "##vso[task.setvariable variable=trtCudaVersion]12.0"
+        - powershell: |
+            Write-Host "##vso[task.setvariable variable=trtCudaVersion;]12.0"
           displayName: Set trtCudaVersion
     - ${{ if and(eq(parameters.CudaVersion, '12.2'), eq(parameters.TrtVersion, '10.0.1.6')) }}:
-        - bash: |
-            echo "##vso[task.setvariable variable=trtCudaVersion]12.4"
+        - powershell: |
+            Write-Host "##vso[task.setvariable variable=trtCudaVersion;]12.4"
           displayName: Set trtCudaVersion
 
-    - bash: |
+    - script: |
         echo $(trtCudaVersion) && echo TensorRT-${{ parameters.TrtVersion }}.Windows10.x86_64.cuda-$(trtCudaVersion)
       displayName: Get trtCudaVersion and Directory Name
 
@@ -63,4 +63,4 @@ steps:
       inputs:
         script: |
           echo %PATH%
-      displayName: 'Print PATH after download TensorRT'
+      displayName: 'Print PATH after download TensorRT'