Commit 69b7a01
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
@@ -44,8 +44,8 @@ RUN apt-get update && apt-get install -y --no-install-recommends \ | |||
| 44 | 44 | ln -s /usr/local/cuda/lib64/stubs/libcuda.so /usr/local/cuda/lib64/stubs/libcuda.so.1 && \ | |
| 45 | 45 | /tmp/clean-layer.sh | |
| 46 | 46 | ||
| 47 | - # Install OpenCL (required by LightGBM GPU version) | ||
| 48 | - RUN apt-get install -y ocl-icd-libopencl1 && \ | ||
| 47 | + # Install OpenCL (required by LightGBM GPU version) and tools. | ||
| 48 | + RUN apt-get install -y ocl-icd-libopencl1 clinfo && \ | ||
| 49 | 49 | /tmp/clean-layer.sh | |
| 50 | 50 | ||
| 51 | 51 | # Install JAX | |
0 commit comments