Spaces:
Building
Building
File size: 1,695 Bytes
bb65e54 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 |
#!/usr/bin/env python3
"""Quick GPU diagnostic for current environment."""
import sys
import os
print("=== GPU Diagnostic ===")
print(f"Python: {sys.version}")
print(f"Platform: {sys.platform}")
print(f"Current directory: {os.getcwd()}")
# Check PyTorch
try:
import torch
print(f"\nPyTorch: {torch.__version__}")
print(f"CUDA available: {torch.cuda.is_available()}")
if torch.cuda.is_available():
print(f"CUDA version: {torch.version.cuda}")
print(f"GPU count: {torch.cuda.device_count()}")
for i in range(torch.cuda.device_count()):
print(f"GPU {i}: {torch.cuda.get_device_name(i)}")
print(f" Memory: {torch.cuda.get_device_properties(i).total_memory / 1024**3:.2f} GB")
else:
print("Running on CPU")
# Check environment variables
print("\nRelevant environment variables:")
for var in ['CUDA_VISIBLE_DEVICES', 'CUDA_HOME', 'SPACES', 'SPACE_ID']:
print(f" {var}: {os.environ.get(var, 'Not set')}")
except ImportError as e:
print(f"PyTorch not available: {e}")
# Check SpaCy
print("\n--- SpaCy Configuration ---")
try:
import spacy
print(f"SpaCy: {spacy.__version__}")
# Try to enable GPU
gpu_id = spacy.prefer_gpu()
print(f"spacy.prefer_gpu(): {gpu_id}")
# Check if we can require GPU
if torch.cuda.is_available():
try:
spacy.require_gpu()
print("spacy.require_gpu(): Success")
except Exception as e:
print(f"spacy.require_gpu(): Failed - {e}")
except ImportError as e:
print(f"SpaCy not available: {e}")
except Exception as e:
print(f"SpaCy error: {e}")
|