File tree 4 files changed +82
-0
lines changed 4 files changed +82
-0
lines changed Original file line number Diff line number Diff line change 1
1
version : 2.1
2
2
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
+
3
10
install_official_git_client : &install_official_git_client
4
11
name : Install Official Git Client
5
12
no_output_timeout : " 1h"
@@ -176,6 +183,20 @@ pytorch_tutorial_build_manager_defaults: &pytorch_tutorial_build_manager_default
176
183
<< : *pytorch_tutorial_build_defaults
177
184
178
185
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
+
179
200
pytorch_tutorial_pr_build_worker_0 :
180
201
<< : *pytorch_tutorial_build_worker_defaults
181
202
Original file line number Diff line number Diff line change
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
Original file line number Diff line number Diff line change
1
+ start /wait " " " %miniconda_exe% " /S /InstallationType=JustMe /RegisterPython=0 /AddToPath=0 /D=%tmp_conda%
Original file line number Diff line number Diff line change
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
You can’t perform that action at this time.
0 commit comments