File tree Expand file tree Collapse file tree 1 file changed +1
-2
lines changed
Expand file tree Collapse file tree 1 file changed +1
-2
lines changed Original file line number Diff line number Diff line change @@ -28,7 +28,7 @@ ARG ARCH
2828ARG CUDA
2929
3030RUN apt-get update \
31- && apt-get -y install ca-certificates gnupg wget libgomp1 build-essential git pkg-config \
31+ && apt-get -y install ca-certificates gnupg wget build-essential git pkg-config \
3232&& wget https://developer.download.nvidia.com/compute/cuda/repos/${DISTRO}/${ARCH}/cuda-keyring_1.1-1_all.deb \
3333&& dpkg -i cuda-keyring_1.1-1_all.deb \
3434&& rm cuda-keyring_1.1-1_all.deb \
@@ -57,7 +57,6 @@ COPY --from=build /usr/local/cuda/lib64/stubs/libcuda.so /usr/local/cuda/lib64/s
5757
5858FROM ${TARGET}
5959ENV LD_LIBRARY_PATH=/usr/local/cuda/lib64:/usr/local/cuda/lib64/stubs
60- ENV PKG_CONFIG_PATH=/usr/lib/pkgconfig
6160ENV PATH=/usr/local/cuda/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin
6261ENV NVIDIA_ARCH="${ARCH}"
6362ENV NVIDIA_CUDA_VERSION="${CUDA}"
You can’t perform that action at this time.
0 commit comments