Skip to content

Commit ebc7f44

Browse files
committed
Add import statement
1 parent 3f66d9a commit ebc7f44

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

test/smoke_test/smoke_test.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -74,6 +74,8 @@ def smoke_test_cuda(package: str) -> None:
7474
print(f"cuDNN enabled? {torch.backends.cudnn.enabled}")
7575

7676
if(package == 'all'):
77+
import torchaudio
78+
import torchvision
7779
if installation_str.find("nightly") != -1:
7880
# just print out cuda version, as version check were already performed during import
7981
print(f"torchvision cuda: {torch.ops.torchvision._cuda_version()}")

0 commit comments

Comments
 (0)