We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 2f902ae commit 738919bCopy full SHA for 738919b
test/smoke_test/smoke_test.py
@@ -81,7 +81,7 @@ def smoke_test_cuda(package: str) -> None:
81
import torchvision
82
# There is an issue with current windows runners calling conda from python
83
# https://github.com/pytorch/test-infra/issues/1054
84
- if installation_str.find("nightly") != -1 or platform.system() == "Windows" :
+ if installation_str.find("nightly") != -1:
85
# just print out cuda version, as version check were already performed during import
86
print(f"torchvision cuda: {torch.ops.torchvision._cuda_version()}")
87
print(f"torchaudio cuda: {torch.ops.torchaudio.cuda_version()}")
0 commit comments