#!/bin/bash
export LD_LIBRARY_PATH=/home/fabian/comfyui-dml/lib/python3.11/site-packages/torch_directml
/home/fabian/comfyui-dml/bin/python -c "
import torch_directml
print('DirectML device:', torch_directml.device())
print('Device name:', torch_directml.device_name(0))
import torch
t = torch.randn(4, 4, device=torch_directml.device())
print('Tensor on DirectML:', t.shape, t.device)
print('SUCCESS')
"
