Skip to content

Commit a176ffe

Browse files
committed
tests/ai-conformance: add toleration for nvidia.com/gpu taint
1 parent 6686205 commit a176ffe

File tree

1 file changed

+10
-0
lines changed

1 file changed

+10
-0
lines changed

tests/e2e/scenarios/ai-conformance/run-test.sh

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -144,13 +144,23 @@ echo "Installing NVIDIA DRA Driver..."
144144
# -f values.yaml \
145145
# --wait
146146

147+
cat > values.yaml <<EOF
148+
# The driver daemonset needs a toleration for the nvidia.com/gpu taint
149+
kubeletPlugin:
150+
tolerations:
151+
- key: nvidia.com/gpu
152+
operator: Exists
153+
effect: NoSchedule
154+
EOF
155+
147156
helm upgrade -i nvidia-dra-driver-gpu nvidia/nvidia-dra-driver-gpu \
148157
--version="25.12.0" \
149158
--create-namespace \
150159
--namespace nvidia-dra-driver-gpu \
151160
--set resources.gpus.enabled=true \
152161
--set nvidiaDriverRoot=/run/nvidia/driver \
153162
--set gpuResourcesEnabledOverride=true \
163+
-f values.yaml \
154164
--wait
155165

156166

0 commit comments

Comments
 (0)