File tree Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Original file line number Diff line number Diff line change @@ -65,7 +65,7 @@ def smoke_test_cuda() -> None:
65
65
66
66
67
67
if installation_str .find ('nightly' ) != - 1 :
68
- # just print out cuda version, as version check were perform during import
68
+ # just print out cuda version, as version check were already performed during import
69
69
print (f"torchvision cuda: { torch .ops .torchvision ._cuda_version ()} " )
70
70
print (f"torchaudio cuda: { torch .ops .torchaudio .cuda_version ()} " )
71
71
else :
@@ -152,9 +152,9 @@ def main() -> None:
152
152
print (f"torch: { torch .__version__ } " )
153
153
print (f"torchvision: { torchvision .__version__ } " )
154
154
print (f"torchaudio: { torchaudio .__version__ } " )
155
+ smoke_test_cuda ()
155
156
if installation_str .find ('nightly' ) != - 1 :
156
157
check_nightly_binaries_date ()
157
- smoke_test_cuda ()
158
158
#smoke_test_conv2d()
159
159
#smoke_test_torchaudio()
160
160
#smoke_test_torchvision()
You can’t perform that action at this time.
0 commit comments