SparseLIDAR-Completion/test_installation.py at main · I2S9/SparseLIDAR-Completion · GitHub

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
"""
Simple test script to verify the installation.
Run: python test_installation.py
"""

print("Testing installation...")
print("=" * 50)

try:
    import torch
    print(f"✓ PyTorch version: {torch.__version__}")
    if torch.cuda.is_available():
        print(f"✓ CUDA available: {torch.version.cuda}")
        print(f"✓ GPU device: {torch.cuda.get_device_name(0)}")
    else:
        print("⚠ CUDA not available (CPU mode)")
except ImportError:
    print("✗ PyTorch not installed")

try:
    import open3d as o3d
    print(f"✓ Open3D version: {o3d.__version__}")
except ImportError:
    print("✗ Open3D not installed")

try:
    import numpy as np
    print(f"✓ NumPy version: {np.__version__}")
except ImportError:
    print("✗ NumPy not installed")

try:
    import scipy
    print(f"✓ SciPy version: {scipy.__version__}")
except ImportError:
    print("✗ SciPy not installed")

try:
    import MinkowskiEngine as ME
    print(f"✓ MinkowskiEngine installed")
except ImportError:
    print("⚠ MinkowskiEngine not installed (install after PyTorch)")

print("=" * 50)
print("Installation test complete!")