File tree Expand file tree Collapse file tree 2 files changed +1
-31
lines changed
Expand file tree Collapse file tree 2 files changed +1
-31
lines changed Original file line number Diff line number Diff line change 8686 condition: endsWith(variables['Agent.JobName'], 'future')
8787 displayName: "extend env. vars 4 future"
8888
89- - bash : |
90- set -e
91- # Try to find nvcc and set CUDA_HOME so that deepspeed can be installed
92- NVCC_PATH=$(find /usr/local -name "nvcc" | head -n 1)
93- if [[ -z "$NVCC_PATH" ]]; then
94- echo "nvcc not found, skipping setting CUDA_HOME"
95- else
96- CUDA_HOME_PATH=$(dirname $(dirname "$NVCC_PATH"))
97- echo "Found nvcc at: ${NVCC_PATH}"
98- echo "Setting CUDA_HOME to: ${CUDA_HOME_PATH}"
99- echo "##vso[task.setvariable variable=CUDA_HOME]${CUDA_HOME_PATH}"
100- fi
101- displayName: "Set CUDA_HOME for DeepSpeed"
102-
10389 - bash : |
10490 echo $(DEVICES)
105- echo $CUDA_HOME_PATH
10691 echo $CUDA_VISIBLE_DEVICES
10792 echo $CUDA_VERSION_MM
10893 echo $TORCH_URL
Original file line number Diff line number Diff line change 9090 condition: endsWith(variables['Agent.JobName'], 'future')
9191 displayName: "extend env. vars 4 future"
9292
93- - bash : |
94- set -e
95- # Try to find nvcc and set CUDA_HOME so that deepspeed can be installed
96- NVCC_PATH=$(find /usr/local -name "nvcc" | head -n 1)
97- if [[ -z "$NVCC_PATH" ]]; then
98- echo "nvcc not found, skipping setting CUDA_HOME"
99- else
100- CUDA_HOME_PATH=$(dirname $(dirname "$NVCC_PATH"))
101- echo "Found nvcc at: ${NVCC_PATH}"
102- echo "Setting CUDA_HOME to: ${CUDA_HOME_PATH}"
103- echo "##vso[task.setvariable variable=CUDA_HOME]${CUDA_HOME_PATH}"
104- fi
105- displayName: "Set CUDA_HOME for DeepSpeed"
106-
10793 - bash : |
10894 echo $(DEVICES)
109- echo $CUDA_HOME_PATH
11095 echo $CUDA_VISIBLE_DEVICES
11196 echo $CUDA_VERSION_MM
11297 echo $TORCH_URL
@@ -227,4 +212,4 @@ jobs:
227212 workingDirectory: examples
228213 env:
229214 PL_USE_MOCKED_MNIST: "1"
230- displayName: "Testing: PyTorch examples"
215+ displayName: "Testing: PyTorch examples"
You can’t perform that action at this time.
0 commit comments