conda 4.10.1

Author: hephaex

python-pytorchlightning/Dockerfile.1.7-py38-cuda11.1 renamed to python-pytorchlightning/Dockerfile.1.7-py38-cuda11.0

@@ -1,17 +1,17 @@
 ARG CUDNN_VERSION=8
-ARG CUDA_VERSION=11.1
-
+ARG CUDA_VERSION=11.0
+ARG CUDA=11
 # FROM nvidia/cuda:${CUDA_VERSION}-devel-ubuntu20.04
 FROM nvidia/cuda:${CUDA_VERSION}-cudnn${CUDNN_VERSION}-devel-ubuntu18.04
 # FROM nvidia/cuda:${CUDA_VERSION}-devel-ubuntu18.04
 
 ARG PYTHON_VERSION=3.8
-ARG PYTORCH_VERSION=1.7
-ARG CONDA_VERSION=4.9.2
+ARG PYTORCH_VERSION=1.7.1
+ARG CONDA_VERSION=4.10.1
 
 SHELL ["/bin/bash", "-c"]
 
-ENV PATH="$PATH:/root/.local/bin"
+#ENV PATH="$PATH:/root/.local/bin"
 
 WORKDIR /opt
 RUN apt-get update -qq && \
@@ -28,9 +28,9 @@ RUN apt-get update -qq && \
 
 # Install conda and python.
 # NOTE new Conda does not forward the exit status... https://github.com/conda/conda/issues/8385
-
-    wget --quiet https://repo.anaconda.com/miniconda/Miniconda3-py38_${CONDA_VERSION}-Linux-x86_64.sh -O miniconda.sh && \
+ 
     mkdir -p /opt && \
+    wget --quiet https://repo.anaconda.com/miniconda/Miniconda3-py38_${CONDA_VERSION}-Linux-x86_64.sh -O miniconda.sh && \
     sh miniconda.sh -b -p /opt/conda && \
     rm miniconda.sh && \
     ln -s /opt/conda/etc/profile.d/conda.sh /etc/profile.d/conda.sh && \
@@ -54,8 +54,8 @@ RUN apt-get update -qq && \
     rm -rf /var/lib/apt/lists/*
 
 ENV \
-    LD_LIBRARY_PATH="/usr/local/cuda/extras/CUPTI/lib64:/usr/local/cuda/compat/lib:/usr/local/nvidia/lib:/usr/local/nvidia/lib64:/usr/local/cuda/lib64:/usr/include/x86_64-linux-gnu:/opt/miniconda3/lib" \
-    PATH="/usr/local/nvidia/bin:/usr/local/cuda/bin:/opt/conda/bin:/usr/local/sbin:/usr/bin/cmake/bin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin:/opt/tensorrt/bin:/opt/miniconda3/bin:$PATH" \
+    LD_LIBRARY_PATH="/usr/local/cuda/extras/CUPTI/lib64:/usr/local/cuda/compat/lib:/usr/local/nvidia/lib:/usr/local/nvidia/lib64:/usr/local/cuda/lib64:/usr/include/x86_64-linux-gnu:/opt/conda/lib" \
+    PATH="/usr/local/nvidia/bin:/usr/local/cuda/bin:/opt/conda/bin:/usr/local/sbin:/usr/bin/cmake/bin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin:/opt/tensorrt/bin:/opt/conda/bin" \
     CUDA_TOOLKIT_ROOT_DIR="/usr/local/cuda" \
     DEBIAN_FRONTEND=noninteractive \
     MPLBACKEND=Svg \
@@ -78,7 +78,7 @@ ENV \
 COPY environment.yml environment.yml
 
 # conda init
-RUN conda create -y --name $CONDA_ENV python=${PYTHON_VERSION} pytorch=${PYTORCH_VERSION} cudatoolkit=${CUDA_VERSION} -c pytorch -c pytorch-test -c pytorch-nightly && \
+RUN conda create -y --name $CONDA_ENV python=${PYTHON_VERSION} pytorch=${PYTORCH_VERSION} cudatoolkit=${CUDA} -c pytorch -c pytorch-test -c pytorch-nightly && \
     conda init bash && \
     # NOTE: this requires that the channel is presented in the yaml before packages
     # replace channel to nigtly if needed, fix PT version and remove Horovod as it will be installed later