Skip to content
Open
Changes from 1 commit
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
9 changes: 6 additions & 3 deletions phases/00-setup-and-tooling/01-dev-environment/code/verify.py
Original file line number Diff line number Diff line change
@@ -1,6 +1,5 @@
import sys
import shutil
import subprocess
import sys

CHECKS = [
("Python 3.10+", lambda: sys.version_info >= (3, 10), f"Python {sys.version}"),
Expand All @@ -17,7 +16,11 @@
(
"CUDA",
lambda: __import__("torch").cuda.is_available(),
lambda: __import__("torch").cuda.get_device_name(0) if __import__("torch").cuda.is_available() else "Not available",
lambda: (
__import__("torch").cuda.get_device_name(0)
if __import__("torch").cuda.is_available()
else "Not available"
),
Comment on lines +19 to +23
),
]

Expand Down
Loading