4 lines
208 B
Python
4 lines
208 B
Python
import torch
|
|
print(f"PyTorch version: {torch.__version__}")
|
|
print(f"CUDA available: {torch.cuda.is_available()}")
|
|
print(f"CUDA version: {torch.version.cuda if torch.cuda.is_available() else 'Not available'}") |