Spaces:
Running
Running
bobqianic
commited on
docker : fix the publishing of the CUDA Docker image (#1704)
Browse files
.devops/main-cuda.Dockerfile
CHANGED
|
@@ -20,6 +20,10 @@ RUN apt-get update && \
|
|
| 20 |
apt-get install -y build-essential \
|
| 21 |
&& rm -rf /var/lib/apt/lists/* /var/cache/apt/archives/*
|
| 22 |
|
|
|
|
|
|
|
|
|
|
|
|
|
| 23 |
COPY .. .
|
| 24 |
RUN make
|
| 25 |
|
|
|
|
| 20 |
apt-get install -y build-essential \
|
| 21 |
&& rm -rf /var/lib/apt/lists/* /var/cache/apt/archives/*
|
| 22 |
|
| 23 |
+
# Ref: https://stackoverflow.com/a/53464012
|
| 24 |
+
ENV CUDA_MAIN_VERSION=12.3
|
| 25 |
+
ENV LD_LIBRARY_PATH /usr/local/cuda-${CUDA_MAIN_VERSION}/compat:$LD_LIBRARY_PATH
|
| 26 |
+
|
| 27 |
COPY .. .
|
| 28 |
RUN make
|
| 29 |
|