2222 " container_image " : " pytorch/conda-builder:cpu" ,
2323 " package_type " : " conda" ,
2424 " build_name " : " conda-py3_7-cpu" ,
25+ " installation " : " conda install pytorch torchvision torchaudio cpuonly -c pytorch-nightly" ,
2526 " validation_runner " : " windows-2019"
2627 },
2728 {
3233 " container_image " : " pytorch/conda-builder:cuda11.3" ,
3334 " package_type " : " conda" ,
3435 " build_name " : " conda-py3_7-cuda11_3" ,
36+ " installation " : " conda install pytorch torchvision torchaudio cpuonly -c pytorch-nightly" ,
3537 " validation_runner " : " windows-2019-m60"
3638 },
3739# {
6870 ]
6971 fail-fast : false
7072 runs-on : ${{ matrix.validation_runner }}
73+ if : matrix.validation_runner == 'windows-2019'
7174 steps :
7275 - name : Checkout PyTorch builder
7376 uses : actions/checkout@v2
8588 if : ${{ matrix.gpu_arch_type == 'cuda' }}
8689 run : |
8790 nvidia-smi
88- - run : conda install --yes numpy
89- - if : matrix.desired_cuda == 'cpu'
90- run : conda install --yes pytorch torchaudio torchvision cpuonly -c pytorch
91- - if : matrix.desired_cuda == 'cu102'
92- run : conda install --yes pytorch torchaudio torchvision cudatoolkit=10.2 -c pytorch
93- - if : matrix.desired_cuda == 'cu113'
94- run : conda install --yes pytorch torchaudio torchvision cudatoolkit=11.3 -c pytorch
95- - if : matrix.desired_cuda == 'cu116'
96- run : conda install --yes pytorch torchaudio torchvision pytorch-cuda=11.6 -c pytorch-nightly -c nvidia
97- - if : matrix.desired_cuda == 'cu117'
98- run : conda intall --yes pytorch torchaudio torchvision pytorch-cuda=11.7 -c pytorch-nightly -c nvidia
91+ - name : Run smoke test non-windows
92+ shell : bash -l {0}
93+ run : ${{ matrix.installation }}
9994 - name : Run smoke test non-windows
10095 shell : bash -l {0}
10196 run : conda run test/smoke_test/smoke_test.py
0 commit comments