Unverified Commit 498b9c86 authored by Andrey Talman's avatar Andrey Talman Committed by GitHub
Browse files

Fix torchvision test for release (#7833)


Co-authored-by: default avatarNicolas Hug <contact@nicolas-hug.com>
parent f4f685dd
...@@ -78,8 +78,13 @@ def smoke_test_torchvision_resnet50_classify(device: str = "cpu") -> None: ...@@ -78,8 +78,13 @@ def smoke_test_torchvision_resnet50_classify(device: str = "cpu") -> None:
def main() -> None: def main() -> None:
print(f"torchvision: {torchvision.__version__}") print(f"torchvision: {torchvision.__version__}")
print(f"torch.cuda.is_available: {torch.cuda.is_available()}") print(f"torch.cuda.is_available: {torch.cuda.is_available()}")
print(f"{torch.ops.image._jpeg_version() = }")
assert torch.ops.image._is_compiled_against_turbo() # Turn 1.11.0aHASH into 1.11 (major.minor only)
version = ".".join(torchvision.__version__.split(".")[:2])
if version >= "0.16":
print(f"{torch.ops.image._jpeg_version() = }")
assert torch.ops.image._is_compiled_against_turbo()
smoke_test_torchvision() smoke_test_torchvision()
smoke_test_torchvision_read_decode() smoke_test_torchvision_read_decode()
smoke_test_torchvision_resnet50_classify() smoke_test_torchvision_resnet50_classify()
......
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