Skip to content

Commit

Permalink
Adding details for the test-infra issue
Browse files Browse the repository at this point in the history
  • Loading branch information
atalman committed Nov 9, 2022
1 parent 0f0f218 commit 2f902ae
Showing 1 changed file with 2 additions and 0 deletions.
2 changes: 2 additions & 0 deletions test/smoke_test/smoke_test.py
Expand Up @@ -79,6 +79,8 @@ def smoke_test_cuda(package: str) -> None:
if(package == 'all'):
import torchaudio
import torchvision
# There is an issue with current windows runners calling conda from python
# https://github.com/pytorch/test-infra/issues/1054
if installation_str.find("nightly") != -1 or platform.system() == "Windows" :
# just print out cuda version, as version check were already performed during import
print(f"torchvision cuda: {torch.ops.torchvision._cuda_version()}")
Expand Down

0 comments on commit 2f902ae

Please sign in to comment.