Skip to content

Commit 21257c3

Browse files
authored
Merge pull request #1311 from dougsland/smi
common: adjust nvidia-smi for check cuda version
2 parents db8f30e + 393a67b commit 21257c3

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

ramalama/common.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -531,7 +531,7 @@ def check_cuda_version():
531531
"""
532532
try:
533533
# Run nvidia-smi --version to get version info
534-
command = ['nvidia-smi', '--version']
534+
command = ['nvidia-smi']
535535
output = run_cmd(command).stdout.decode("utf-8").strip()
536536

537537
# Look for CUDA Version in the output

0 commit comments

Comments
 (0)