Skip to content

Commit 83b580c

Browse files
committed
[test]test for windows build
1 parent 43e3eb7 commit 83b580c

File tree

4 files changed

+82
-0
lines changed

4 files changed

+82
-0
lines changed

.circleci/config.yml

Lines changed: 21 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,12 @@
11
version: 2.1
22

3+
executors:
4+
windows-with-nvidia-gpu:
5+
machine:
6+
resource_class: windows.gpu.nvidia.medium
7+
image: windows-server-2019-nvidia:stable
8+
shell: bash.exe
9+
310
install_official_git_client: &install_official_git_client
411
name: Install Official Git Client
512
no_output_timeout: "1h"
@@ -176,6 +183,20 @@ pytorch_tutorial_build_manager_defaults: &pytorch_tutorial_build_manager_default
176183
<<: *pytorch_tutorial_build_defaults
177184

178185
jobs:
186+
pytorch_windows_build:
187+
executor: windows-with-nvidia-gpu
188+
steps:
189+
- checkout
190+
- run:
191+
name: Install Cuda
192+
no_output_timeout: 30m
193+
command: |
194+
.circleci/scripts/windows_cuda_install.sh
195+
- run:
196+
name: test
197+
command: |
198+
.circleci/scripts/build_for_windows.sh
199+
179200
pytorch_tutorial_pr_build_worker_0:
180201
<<: *pytorch_tutorial_build_worker_defaults
181202

Lines changed: 23 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,23 @@
1+
#!/bin/bash
2+
set -eux -o pipefail
3+
4+
SOURCE_DIR="$( cd "$( dirname "${BASH_SOURCE[0]}" )" >/dev/null && pwd )"
5+
pushd $SOURCE_DIR
6+
7+
#install anaconda3
8+
export CONDA_HOME="${SOURCE_DIR}/conda"
9+
export tmp_conda="${SOURCE_DIR}/conda"
10+
export miniconda_exe="${SOURCE_DIR}/miniconda.exe"
11+
rm -rf conda miniconda.exe
12+
curl -k https://repo.anaconda.com/miniconda/Miniconda3-latest-Windows-x86_64.exe -o miniconda.exe
13+
./install_conda.bat
14+
export PATH="$(tmp_conda):$(tmp_conda)/Library/usr/bin:$(tmp_conda)/Library/bin:$(tmp_conda)/Scripts:$(tmp_conda)/bin:$PATH"
15+
16+
conda create -qyn testenv python=3.7
17+
conda activate testenv
18+
19+
#conda install -yq -c pytorch-nightly "cudatoolkit=%CUDA_VERSION_STR%" %package_name_and_version%
20+
conda install -r ../../requirements.txt
21+
22+
cd ../..
23+
make docs

.circleci/scripts/install_conda.bat

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1 @@
1+
start /wait "" "%miniconda_exe%" /S /InstallationType=JustMe /RegisterPython=0 /AddToPath=0 /D=%tmp_conda%
Lines changed: 37 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,37 @@
1+
#!/bin/bash
2+
set -eux -o pipefail
3+
4+
curl --retry 3 -kLO https://ossci-windows.s3.amazonaws.com/cuda_10.1.243_426.00_win10.exe
5+
7z x cuda_10.1.243_426.00_win10.exe -ocuda_10.1.243_426.00_win10
6+
cd cuda_10.1.243_426.00_win10
7+
mkdir cuda_install_logs
8+
9+
set +e
10+
11+
./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"
12+
13+
set -e
14+
15+
if [[ "${VC_YEAR}" == "2017" ]]; then
16+
cp -r CUDAVisualStudioIntegration/extras/visual_studio_integration/MSBuildExtensions/* "C:/Program Files (x86)/Microsoft Visual Studio/2017/${VC_PRODUCT}/Common7/IDE/VC/VCTargets/BuildCustomizations/"
17+
else
18+
cp -r CUDAVisualStudioIntegration/extras/visual_studio_integration/MSBuildExtensions/* "C:/Program Files (x86)/Microsoft Visual Studio/2019/${VC_PRODUCT}/MSBuild/Microsoft/VC/v160/BuildCustomizations/"
19+
fi
20+
21+
curl --retry 3 -kLO https://ossci-windows.s3.amazonaws.com/NvToolsExt.7z
22+
7z x NvToolsExt.7z -oNvToolsExt
23+
mkdir -p "C:/Program Files/NVIDIA Corporation/NvToolsExt"
24+
cp -r NvToolsExt/* "C:/Program Files/NVIDIA Corporation/NvToolsExt/"
25+
export NVTOOLSEXT_PATH="C:\\Program Files\\NVIDIA Corporation\\NvToolsExt\\"
26+
27+
if ! ls "/c/Program Files/NVIDIA GPU Computing Toolkit/CUDA/v10.1/bin/nvcc.exe"
28+
then
29+
echo "CUDA installation failed"
30+
mkdir -p /c/w/build-results
31+
7z a "c:\\w\\build-results\\cuda_install_logs.7z" cuda_install_logs
32+
exit 1
33+
fi
34+
35+
cd ..
36+
rm -rf ./cuda_10.1.243_426.00_win10
37+
rm -f ./cuda_10.1.243_426.00_win10.exe

0 commit comments

Comments
 (0)