Uploaded sanitized BC250/ROCm Repository.
This commit is contained in:
@@ -0,0 +1,11 @@
|
||||
#!/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')
|
||||
"
|
||||
Reference in New Issue
Block a user