Merge remote-tracking branch 'origin/ray-tune' into ray-tune

Author: Kai Fricke

.circleci/config.yml

@@ -1,5 +1,12 @@
 version: 2.1
 
+executors:
+  windows-with-nvidia-gpu:
+    machine:
+      resource_class: windows.gpu.nvidia.medium
+      image: windows-server-2019-nvidia:stable
+      shell: bash.exe
+
 install_official_git_client: &install_official_git_client
   name: Install Official Git Client
   no_output_timeout: "1h"
@@ -175,6 +182,20 @@ pytorch_tutorial_build_manager_defaults: &pytorch_tutorial_build_manager_default
   resource_class: medium
   <<: *pytorch_tutorial_build_defaults
 
+pytorch_windows_build: &pytorch_windows_build
+  executor: windows-with-nvidia-gpu
+  steps:
+    - checkout
+    - run:
+        name: Install Cuda
+        no_output_timeout: 30m
+        command: |
+          .circleci/scripts/windows_cuda_install.sh
+    - run:
+        name: test
+        command: |
+          .circleci/scripts/build_for_windows.sh
+
 jobs:
   pytorch_tutorial_pr_build_worker_0:
     <<: *pytorch_tutorial_build_worker_defaults
@@ -302,6 +323,9 @@ jobs:
   pytorch_tutorial_master_build_manager:
     <<: *pytorch_tutorial_build_manager_defaults
 
+  pytorch_windows_build_worker:
+    <<: *pytorch_windows_build
+
 workflows:
   build:
     jobs:
@@ -538,3 +562,10 @@ workflows:
             branches:
               only:
                 - master
+      - pytorch_windows_build_worker:
+          name: win_test_worker
+          filters:
+            branches:
+              only:
+                - master
+

.circleci/scripts/build_for_windows.sh

@@ -0,0 +1,49 @@
+#!/bin/bash
+set -eux -o pipefail
+
+retry () {
+  $*  || (sleep 1 && $*) || (sleep 2 && $*)
+}
+
+SOURCE_DIR="$( cd "$( dirname "${BASH_SOURCE[0]}" )" >/dev/null && pwd )"
+PROJECT_DIR="${SOURCE_DIR}/../.."
+pushd $SOURCE_DIR
+
+#install wget and make
+curl -k https://ymu.dl.osdn.jp/mingw/68260/mingw-get-0.6.3-mingw32-pre-20170905-1-bin.zip -o mingw32.zip
+unzip mingw32.zip -d mingw32
+mingw32/bin/mingw-get.exe install mingw32-make
+mingw32/bin/mingw-get.exe install msys-findutils
+mv mingw32/bin/mingw32-make.exe mingw32/bin/make.exe
+curl -k https://eternallybored.org/misc/wget/1.20.3/64/wget.exe -o mingw32/bin/wget.exe
+export PATH="${SOURCE_DIR}/mingw32/bin:$PATH"
+
+#install anaconda3
+export CONDA_HOME="${SOURCE_DIR}/conda"
+export tmp_conda="${SOURCE_DIR}/conda"
+export miniconda_exe="${SOURCE_DIR}/miniconda.exe"
+rm -rf conda miniconda.exe
+curl -k https://repo.anaconda.com/miniconda/Miniconda3-latest-Windows-x86_64.exe -o miniconda.exe
+./install_conda.bat
+export PATH="${tmp_conda}:${tmp_conda}/Library/usr/bin:${tmp_conda}/Library/bin:${tmp_conda}/Scripts:${tmp_conda}/bin:$PATH"
+
+eval "$(conda shell.bash hook)"
+conda create -qyn testenv python=3.7
+conda activate testenv
+
+conda install sphinx
+pip install sphinx_gallery==0.3.1 flask pandas spacy ipython scipy pySoundFile scikit-image
+pip install -e git+git://github.com/pytorch/pytorch_sphinx_theme.git#egg=pytorch_sphinx_theme
+conda install -yq -c pytorch "cudatoolkit=10.1" pytorch torchvision torchtext
+conda install torchaudio -c pytorch-test
+python -m spacy download de
+python -m spacy download en
+pushd ${PROJECT_DIR}
+DIR=.jenkins
+python $DIR/remove_runnable_code.py beginner_source/aws_distributed_training_tutorial.py beginner_source/aws_distributed_training_tutorial.py || true
+python $DIR/remove_runnable_code.py beginner_source/data_loading_tutorial.py beginner_source/data_loading_tutorial.py || true
+python $DIR/remove_runnable_code.py beginner_source/dcgan_faces_tutorial.py beginner_source/dcgan_faces_tutorial.py || true
+python $DIR/remove_runnable_code.py intermediate_source/model_parallel_tutorial.py intermediate_source/model_parallel_tutorial.py || true
+python $DIR/remove_runnable_code.py intermediate_source/memory_format_tutorial.py intermediate_source/memory_format_tutorial.py || true
+
+make docs

.circleci/scripts/install_conda.bat

@@ -0,0 +1 @@
+start /wait "" "%miniconda_exe%" /S /InstallationType=JustMe /RegisterPython=0 /AddToPath=0 /D=%tmp_conda%

.circleci/scripts/windows_cuda_install.sh

@@ -0,0 +1,31 @@
+#!/bin/bash
+set -eux -o pipefail
+
+curl --retry 3 -kLO https://ossci-windows.s3.amazonaws.com/cuda_10.1.243_426.00_win10.exe
+7z x cuda_10.1.243_426.00_win10.exe -ocuda_10.1.243_426.00_win10
+cd cuda_10.1.243_426.00_win10
+mkdir cuda_install_logs
+
+set +e
+
+./setup.exe -s nvcc_10.1 cuobjdump_10.1 nvprune_10.1 cupti_10.1 cublas_10.1 cublas_dev_10.1 cudart_10.1 cufft_10.1 cufft_dev_10.1 curand_10.1 curand_dev_10.1 cusolver_10.1 cusolver_dev_10.1 cusparse_10.1 cusparse_dev_10.1 nvgraph_10.1 nvgraph_dev_10.1 npp_10.1 npp_dev_10.1 nvrtc_10.1 nvrtc_dev_10.1 nvml_dev_10.1 -loglevel:6 -log:"$(pwd -W)/cuda_install_logs"
+
+set -e
+
+curl --retry 3 -kLO https://ossci-windows.s3.amazonaws.com/NvToolsExt.7z
+7z x NvToolsExt.7z -oNvToolsExt
+mkdir -p "C:/Program Files/NVIDIA Corporation/NvToolsExt"
+cp -r NvToolsExt/* "C:/Program Files/NVIDIA Corporation/NvToolsExt/"
+export NVTOOLSEXT_PATH="C:\\Program Files\\NVIDIA Corporation\\NvToolsExt\\"
+
+if ! ls "/c/Program Files/NVIDIA GPU Computing Toolkit/CUDA/v10.1/bin/nvcc.exe"
+then
+    echo "CUDA installation failed"
+    mkdir -p /c/w/build-results
+    7z a "c:\\w\\build-results\\cuda_install_logs.7z" cuda_install_logs
+    exit 1
+fi
+
+cd ..
+rm -rf ./cuda_10.1.243_426.00_win10
+rm -f ./cuda_10.1.243_426.00_win10.exe

.jenkins/build.sh

@@ -15,8 +15,8 @@ export PATH=/opt/conda/bin:$PATH
 rm -rf src
 pip install -r $DIR/../requirements.txt
 
-export PATH=/opt/conda/bin:$PATH
-pip install sphinx==1.8.2 pandas
+# export PATH=/opt/conda/bin:$PATH
+# pip install sphinx==1.8.2 pandas
 
 # For Tensorboard. Until 1.14 moves to the release channel.
 pip install tb-nightly

Makefile

@@ -43,6 +43,7 @@ download:
 	mkdir -p advanced_source/data
 	mkdir -p beginner_source/data
 	mkdir -p intermediate_source/data
+	mkdir -p prototype_source/data
 
 	# transfer learning tutorial data
 	wget -N https://download.pytorch.org/tutorial/hymenoptera_data.zip -P $(DATADIR)
@@ -100,6 +101,14 @@ download:
 	wget -N https://s3.amazonaws.com/pytorch-tutorial-assets/imagenet_1k.zip -P $(DATADIR)
 	unzip $(ZIPOPTS) $(DATADIR)/imagenet_1k.zip -d advanced_source/data/
 
+	# Download model for prototype_source/graph_mode_static_quantization_tutorial.py
+	wget -N https://download.pytorch.org/models/resnet18-5c106cde.pth -P $(DATADIR)
+	cp $(DATADIR)/resnet18-5c106cde.pth prototype_source/data/resnet18_pretrained_float.pth
+
+	# Download dataset for prototype_source/graph_mode_static_quantization_tutorial.py
+	wget -N https://s3.amazonaws.com/pytorch-tutorial-assets/imagenet_1k.zip -P $(DATADIR)
+	unzip $(ZIPOPTS) $(DATADIR)/imagenet_1k.zip -d prototype_source/data/
+
 docs:
 	make download
 	make html