# === Fame Kit Test Script === | |
set -e | |
echo "🧪 Importing torch and checking ROCm support..." | |
python3 -c " | |
import torch | |
print('Torch Version:', torch.__version__) | |
print('CUDA available:', torch.cuda.is_available()) | |
print('HIP version (ROCm):', getattr(torch.version, 'hip', 'N/A')) | |
print('Built with ROCm:', torch.version.hip is not None) | |
" | |
echo "✅ Test completed!" | |