Fabian
23 lines
759 B
Bash
23 lines
759 B
Bash
#!/bin/bash
|
|
/home/fabian/ComfyUI/venv/bin/python -c "
|
|
import torch
|
|
print('CUDA available:', torch.cuda.is_available())
|
|
print('HIP version:', torch.version.hip)
|
|
if torch.cuda.is_available():
|
|
print('Device:', torch.cuda.get_device_name(0))
|
|
print('VRAM:', torch.cuda.get_device_properties(0).total_memory / 1024**3, 'GB')
|
|
# Quick matmul test
|
|
import time
|
|
a = torch.randn(1024, 1024, device='cuda', dtype=torch.float16)
|
|
b = torch.randn(1024, 1024, device='cuda', dtype=torch.float16)
|
|
torch.cuda.synchronize()
|
|
t0 = time.time()
|
|
for _ in range(10):
|
|
c = torch.matmul(a, b)
|
|
torch.cuda.synchronize()
|
|
t1 = time.time()
|
|
print(f'10x matmul 1024x1024 f16: {(t1-t0)*1000:.0f}ms')
|
|
else:
|
|
print('CUDA NOT AVAILABLE')
|
|
"
|