how to run it in GPU？？？

[herbertchen1]

i run pip install -r req....

but ,couldnt run it with python main.py --cuda

with the trace back:
AssertionError: Torch not compiled with CUDA enabled

[dasguptar]

Hi @herbertchen1 , the requirements.txt file contains the CPU version of PyTorch. You can go to https://pytorch.org/ and get the URL for the wheel file for the GPU version, and install that version.

[herbertchen1]

Thanks very much！！！