diff --git a/Dockerfile.cuda b/Dockerfile.cuda index bb8909a..a6d4ac4 100755 --- a/Dockerfile.cuda +++ b/Dockerfile.cuda @@ -28,7 +28,7 @@ ARG ARCH ARG CUDA RUN apt-get update \ -&& apt-get -y install ca-certificates gnupg wget libgomp1 build-essential git pkg-config \ +&& apt-get -y install ca-certificates gnupg wget build-essential git pkg-config \ && wget https://developer.download.nvidia.com/compute/cuda/repos/${DISTRO}/${ARCH}/cuda-keyring_1.1-1_all.deb \ && dpkg -i cuda-keyring_1.1-1_all.deb \ && 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 FROM ${TARGET} ENV LD_LIBRARY_PATH=/usr/local/cuda/lib64:/usr/local/cuda/lib64/stubs -ENV PKG_CONFIG_PATH=/usr/lib/pkgconfig ENV PATH=/usr/local/cuda/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin ENV NVIDIA_ARCH="${ARCH}" ENV NVIDIA_CUDA_VERSION="${CUDA}"