Fabian
12 lines
405 B
Bash
12 lines
405 B
Bash
#!/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')
|
|
"
|