search

dasguptar / treelstm.pytorch

Tree LSTM implementation in PyTorch
MIT License
550 stars 139 forks source link

how to run it in GPU？？？ #27

Closed herbertchen1 closed 6 years ago

herbertchen1 commented 6 years ago

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 commented 6 years ago

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 commented 6 years ago

Thanks very much！！！