We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 3f66d9a commit ebc7f44Copy full SHA for ebc7f44
test/smoke_test/smoke_test.py
@@ -74,6 +74,8 @@ def smoke_test_cuda(package: str) -> None:
74
print(f"cuDNN enabled? {torch.backends.cudnn.enabled}")
75
76
if(package == 'all'):
77
+ import torchaudio
78
+ import torchvision
79
if installation_str.find("nightly") != -1:
80
# just print out cuda version, as version check were already performed during import
81
print(f"torchvision cuda: {torch.ops.torchvision._cuda_version()}")
0 commit comments