Fixed some missing CUDA old version

tlkh · Jan 6, 2020 · dab8b0d · dab8b0d
1 parent 1665efe
commit dab8b0d
Show file tree

Hide file tree

Showing 9 changed files with 16 additions and 712 deletions.
diff --git a/src/base-batch.Dockerfile b/src/base-batch.Dockerfile
@@ -161,7 +161,7 @@ RUN apt-get update && \
     cuda-command-line-tools-$CUDA_PKG_VERSION \
     cuda-libraries-dev-$CUDA_PKG_VERSION \
     cuda-minimal-build-$CUDA_PKG_VERSION \
-    libnccl-dev=$NCCL_VERSION-1+cuda10.0 && \
+    libnccl-dev=$NCCL_VERSION-1+cuda10.1 && \
     cd /tmp/ && \
     git clone --depth 1 https://github.com/NVIDIA/apex && \
     cd apex && \
@@ -183,8 +183,8 @@ RUN apt-get update && \
 USER $NB_UID
 
 RUN conda install \
-      -c nvidia/label/cuda10.0 \
-      -c rapidsai/label/cuda10.0 \
+      -c nvidia/label/cuda10.1 \
+      -c rapidsai/label/cuda10.1 \
       -c numba -c conda-forge -c defaults \
       'python=3.6' \
       'numpy=1.16.1' \

diff --git a/src/build-tf.Dockerfile b/src/build-tf.Dockerfile
@@ -14,7 +14,7 @@ RUN apt-get update && \
     cuda-command-line-tools-$CUDA_PKG_VERSION \
     cuda-libraries-dev-$CUDA_PKG_VERSION \
     cuda-minimal-build-$CUDA_PKG_VERSION \
-    libnccl-dev=$NCCL_VERSION-1+cuda10.0 \
+    libnccl-dev=$NCCL_VERSION-1+cuda10.1 \
     python3-dev \
     python3-numpy \
     python3-six \

diff --git a/src/full.Dockerfile b/src/full.Dockerfile
@@ -17,7 +17,7 @@ RUN conda install -c pytorch --quiet --yes \
       'numpy=1.16.1' \
       'pytorch' \
       'torchvision' \
-      'cudatoolkit=10.0' && \
+      'cudatoolkit=10.1' && \
     pip install --no-cache-dir torchtext && \
     conda install -c pytorch -c fastai --quiet --yes \
       'python=3.6' \
@@ -48,7 +48,7 @@ RUN apt-get update && \
     cuda-command-line-tools-$CUDA_PKG_VERSION \
     cuda-libraries-dev-$CUDA_PKG_VERSION \
     cuda-minimal-build-$CUDA_PKG_VERSION \
-    libnccl-dev=$NCCL_VERSION-1+cuda10.0 && \
+    libnccl-dev=$NCCL_VERSION-1+cuda10.1 && \
     cd /tmp/ && \
     git clone --depth 1 https://github.com/NVIDIA/apex && \
     cd apex && \
@@ -60,7 +60,7 @@ RUN apt-get update && \
       cuda-nvml-dev-$CUDA_PKG_VERSION \
       cuda-command-line-tools-$CUDA_PKG_VERSION \
       cuda-libraries-dev-$CUDA_PKG_VERSION \
-      libnccl-dev=$NCCL_VERSION-1+cuda10.0 && \
+      libnccl-dev=$NCCL_VERSION-1+cuda10.1 && \
     apt-get autoremove -y && \
     apt-get clean && \
     rm -rf /var/lib/apt/lists/* /var/cache/apt/* && \
@@ -75,8 +75,8 @@ RUN apt-get update && \
 USER $NB_UID
 
 RUN conda install \
-      -c nvidia/label/cuda10.0 \
-      -c rapidsai/label/cuda10.0 \
+      -c nvidia/label/cuda10.1 \
+      -c rapidsai/label/cuda10.1 \
       -c numba -c conda-forge -c defaults \
       'python=3.6' \
       'numpy=1.16.1' \
@@ -121,13 +121,13 @@ RUN apt-get update && \
       cuda-command-line-tools-$CUDA_PKG_VERSION \
       cuda-libraries-dev-$CUDA_PKG_VERSION \
       cuda-minimal-build-$CUDA_PKG_VERSION \
-      libnccl-dev=$NCCL_VERSION-1+cuda10.0 && \
+      libnccl-dev=$NCCL_VERSION-1+cuda10.1 && \
     pip install --no-cache-dir horovod && \
     apt-get remove -yq \
       cuda-nvml-dev-$CUDA_PKG_VERSION \
       cuda-command-line-tools-$CUDA_PKG_VERSION \
       cuda-libraries-dev-$CUDA_PKG_VERSION \
-      libnccl-dev=$NCCL_VERSION-1+cuda10.0 && \
+      libnccl-dev=$NCCL_VERSION-1+cuda10.1 && \
     apt-get autoremove -y && \
     apt-get clean && \
     rm -rf /var/lib/apt/lists/* /var/cache/apt/* && \

diff --git a/src/old/old.rapids.Dockerfile b/src/old/old.rapids.Dockerfile
diff --git a/src/old/old.tf-amp.Dockerfile b/src/old/old.tf-amp.Dockerfile
diff --git a/src/old/start-rapids.sh b/src/old/start-rapids.sh