1ytic / pytorch-edit-distance

Levenshtein edit-distance on PyTorch and CUDA
MIT License
94 stars 14 forks source link

Does it supports torch CPU? #7

Closed KinWaiCheuk closed 3 years ago

KinWaiCheuk commented 3 years ago

I have a machine with only CPU for debugging. However, I am unable to install your library via pip install torch_edit_distance. The following errors appear

OSError: CUDA_HOME environment variable is not set. Please set it to your CUDA install root. ERROR: Could not find a version that satisfies the requirement torch_edit_distance ERROR: No matching distribution found for torch_edit_distance

1ytic commented 3 years ago

Unfortunately, you can't install it on a machine without GPU.