【iluvatar】fix docker bug (#320)

* fix the error that cannot generate docker image

* fix iluvatar docker bug

* fix the spelling error

* add 'apt install -y libncursesw5'

training/iluvatar/docker_image/pytorch/Dockerfile

@@ -1,4 +1,7 @@
-FROM ubuntu:18.04
+FROM ubuntu:20.04
+
+# copy /etc/apt/sources.list . or choose an available one if encountering a problem with the mirror source
+# ADD sources.list /etc/apt/
 
 RUN /bin/bash -c "source /root/.bashrc"
 
@@ -33,11 +36,14 @@ RUN apt-get install -y --fix-missing \
      pbzip2 \
      pv \
      numactl \
-     ninja-build 
+     ninja-build \
+     gcc-7 \
+     g++-7 \
+     libncursesw5
 
 
 # Configure anaconda
-RUN wget https://repo.anaconda.com/miniconda/Miniconda3-py38_4.10.3-Linux-x86_64.sh && \
+RUN wget https://mirrors.tuna.tsinghua.edu.cn/anaconda/miniconda/Miniconda3-py38_4.10.3-Linux-x86_64.sh && \
     bash ./Miniconda3-py38_4.10.3-Linux-x86_64.sh -b -p /root/miniconda && \
     /root/miniconda/bin/conda clean -tipsy && \
     ln -s /root/miniconda/etc/profile.d/conda.sh /etc/profile.d/conda.sh && \
@@ -50,6 +56,8 @@ RUN wget https://repo.anaconda.com/miniconda/Miniconda3-py38_4.10.3-Linux-x86_64
 
 RUN /bin/bash -c "apt-get install -y linux-headers-`uname -r`"
 
+RUN update-alternatives --install /usr/bin/gcc gcc /usr/bin/gcc-7 10 --slave /usr/bin/g++ g++ /usr/bin/g++-7
+
 RUN /bin/bash -c "pip config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple"
 
 ENV LD_LIBRARY_PATH="/usr/local/corex/lib:${LD_LIBRARY_PATH}"

training/iluvatar/docker_image/pytorch/pytorch_install.sh

@@ -13,9 +13,14 @@ done
 search_sdk_results=`find ${SDK_DIR} -name "corex*.run"`
 for installer in $search_sdk_results; do
     echo "Install ${installer}"
-    sh "${installer}" -- --silent --driver --toolkit
+    sh "${installer}" -- --silent --toolkit
 done
 
+torch_packages_results=`find ${PKG_DIR} -name "torch-*.whl"`
+if [ -n "$torch_packages_results" ]; then    
+    pip3 install "$torch_packages_results"
+fi
+
 search_packages_results=`find ${PKG_DIR} -name "*.whl"`
 for pkg in $search_packages_results; do
     echo "Install ${pkg}"