Skip to content

Commit b96c494

Browse files
committed
fixed cuda min resources
1 parent 219ff15 commit b96c494

File tree

3 files changed

+4
-5
lines changed

3 files changed

+4
-5
lines changed

commons/Dockerfile.base.20.03-py36-cuda10.1

Lines changed: 4 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -36,12 +36,11 @@ RUN apt-get update -y && \
3636
vim \
3737
yasm \
3838
wget zip unzip \
39-
ca-certificates \
4039
software-properties-common \
4140
openssh-client openssh-server \
4241
libssl-dev \
4342
libmpdec2 \
44-
pdsh curl sudo net-tools \
43+
pdsh curl net-tools \
4544
iputils-ping \
4645
cuda-command-line-tools-${CUDA/./-} \
4746
libcublas10=10.2.1.243-1 \
@@ -130,8 +129,8 @@ RUN mkdir -p ${STAGE_DIR} && \
130129
dpkg-buildpackage -us -uc && \
131130
dpkg -i ${STAGE_DIR}/nvidia-peer-memory_1.0-8_all.deb
132131

133-
# Install CUDA-10.0 + cuDNN 7.6.0
134-
RUN ln -s /usr/local/cuda-10.0 /usr/local/cuda && \
132+
# Install CUDA-10.1 + cuDNN 7.6.0
133+
RUN ln -s /usr/local/cuda-10.1 /usr/local/cuda && \
135134
ln -s /usr/lib/x86_64-linux-gnu/libcudnn.so.7.6.0 /usr/local/cuda/lib64/libcudnn.so && \
136135
ldconfig
137136

@@ -319,4 +318,4 @@ RUN apt autoclean && \
319318
rm -rf /root/.cache && \
320319
rm -rf /tmp/*
321320

322-
WORKDIR /home/work
321+
WORKDIR /home/work

0 commit comments

Comments
 (0)