bobqianic commited on
Commit
6091193
·
unverified ·
1 Parent(s): 397f291

docker : fix the publishing of the CUDA Docker image (#1704)

Browse files
Files changed (1) hide show
  1. .devops/main-cuda.Dockerfile +4 -0
.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