Unverified Commit e5a5f0be authored by Nikita Shulga's avatar Nikita Shulga Committed by GitHub
Browse files

Update hardcoded WIN-CUDA from 11.1 with 11.3 (#5451)

Also, remove reference to conda-forge, all CUDA toolchain should be
available from NVIDIA channel

Install h5py from pip on Windows and skip failing gaussian_blur tests if Win+CUDA11.3 is used
parent 86a14cba
......@@ -848,7 +848,7 @@ jobs:
executor:
name: windows-gpu
environment:
CUDA_VERSION: "11.1"
CUDA_VERSION: "11.3"
PYTHON_VERSION: << parameters.python_version >>
steps:
- checkout
......
......@@ -848,7 +848,7 @@ jobs:
executor:
name: windows-gpu
environment:
CUDA_VERSION: "11.1"
CUDA_VERSION: "11.3"
PYTHON_VERSION: << parameters.python_version >>
steps:
- checkout
......
......@@ -9,10 +9,11 @@ dependencies:
- libpng
- jpeg
- ca-certificates
- h5py
- hdf5
- pip:
- future
- pillow >=5.3.0, !=8.3.*
- scipy
- av
- dataclasses
- h5py
......@@ -28,7 +28,7 @@ else
fi
printf "Installing PyTorch with %s\n" "${cudatoolkit}"
conda install -y -c "pytorch-${UPLOAD_CHANNEL}" -c conda-forge "pytorch-${UPLOAD_CHANNEL}"::pytorch[build="*${version}*"] "${cudatoolkit}" pytest
conda install -y -c "pytorch-${UPLOAD_CHANNEL}" -c nvidia "pytorch-${UPLOAD_CHANNEL}"::pytorch[build="*${version}*"] "${cudatoolkit}" pytest
torch_cuda=$(python -c "import torch; print(torch.cuda.is_available())")
echo torch.cuda.is_available is $torch_cuda
......
import os
import sys
import numpy as np
import pytest
......@@ -958,6 +959,17 @@ def test_random_apply(device):
)
@pytest.mark.parametrize("channels", [1, 3])
def test_gaussian_blur(device, channels, meth_kwargs):
if all(
[
device == "cuda",
channels == 1,
meth_kwargs["kernel_size"] in [23, [23]],
torch.version.cuda == "11.3",
sys.platform in ("win32", "cygwin"),
]
):
pytest.skip("Fails on Windows, see https://github.com/pytorch/vision/issues/5464")
tol = 1.0 + 1e-10
torch.manual_seed(12)
_test_class_op(
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment