I tried to use GPU to run SCALE by this commond
SCALE.py -d a.txt -k 20 --binary --gpu 1
But I got this error:
Traceback (most recent call last):
File "/opt/conda/bin/SCALE.py", line 71, in <module>
torch.cuda.set_device(args.gpu)
File "/opt/conda/lib/python3.8/site-packages/torch/cuda/__init__.py", line 263, in set_device
torch._C._cuda_setDevice(device)
RuntimeError: CUDA error: invalid device ordinal
Could you please tell me how to use GPU correctly to run SCALE? Many thanks!
I tried to use GPU to run SCALE by this commond
SCALE.py -d a.txt -k 20 --binary --gpu 1
But I got this error:Could you please tell me how to use GPU correctly to run SCALE? Many thanks!