| commit | 48c23ac0b94fc2c757069f094446489e18b1e329 | [log] [tgz] |
|---|---|---|
| author | Vitaly Terentyev <vitaly.terentyev@akvelon.com> | Sat Sep 27 11:11:27 2025 +0400 |
| committer | Vitaly Terentyev <vitaly.terentyev@akvelon.com> | Sat Sep 27 11:11:27 2025 +0400 |
| tree | f7507856e046b1ed965d816548e729d232dd8fda | |
| parent | 8ae10c43788b866032884e0276fee6c0a836155d [diff] |
Use 12.2.2 cuda
diff --git a/sdks/python/apache_beam/ml/inference/test_resources/vllm.dockerfile b/sdks/python/apache_beam/ml/inference/test_resources/vllm.dockerfile index 0fe0809..b839838 100644 --- a/sdks/python/apache_beam/ml/inference/test_resources/vllm.dockerfile +++ b/sdks/python/apache_beam/ml/inference/test_resources/vllm.dockerfile
@@ -17,7 +17,7 @@ # Used for any vLLM integration test # Dockerfile — Beam dev harness + install dev SDK from LOCAL source package -FROM nvidia/cuda:12.4.1-devel-ubuntu22.04 +FROM nvidia/cuda:12.2.2-devel-ubuntu22.04 # 1) Non-interactive + timezone ENV DEBIAN_FRONTEND=noninteractive \