Skip to content

GitLab

Commit a4714569 authored by Matthew Douglas's avatar Matthew Douglas
Browse files

fix formatting for install_cuda.py

parent c17fb8eb
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 6 deletions
install_cuda.py
View file @ a4714569
...@@ -77,9 +77,7 @@ def main(): ...@@ -77,9 +77,7 @@ def main():
download_path = "/tmp" # default download path download_path = "/tmp" # default download path
if len(sys.argv) < 2: if len(sys.argv) < 2:
print( print("Usage: python install_cuda.py <version/all> [user/system] [download_path]")
"Usage: python install_cuda.py <version/all> [user/system] [download_path]"
)
sys.exit(1) sys.exit(1)
version = sys.argv[1] version = sys.argv[1]
...@@ -100,9 +98,7 @@ def main(): ...@@ -100,9 +98,7 @@ def main():
elif version in cuda_versions: elif version in cuda_versions:
install_cuda(version, base_path, download_path) install_cuda(version, base_path, download_path)
else: else:
print( print(f"Invalid CUDA version: {version}. Available versions are: {', '.join(cuda_versions.keys())}")
f"Invalid CUDA version: {version}. Available versions are: {', '.join(cuda_versions.keys())}"
)
sys.exit(1) sys.exit(1)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment