| import torch | |
| print("PyTorch version:", torch.__version__) | |
| print("CUDA available:", torch.cuda.is_available()) | |
| print("CUDA compiled version:", torch.version.cuda) | |
| print("GPU count:", torch.cuda.device_count()) |
| import torch | |
| print("PyTorch version:", torch.__version__) | |
| print("CUDA available:", torch.cuda.is_available()) | |
| print("CUDA compiled version:", torch.version.cuda) | |
| print("GPU count:", torch.cuda.device_count()) |