search

koursaros-ai / nboost

NBoost is a scalable, search-api-boosting platform for deploying transformer models to improve the relevance of search results on different platforms (i.e. Elasticsearch)
Apache License 2.0
675 stars 69 forks source link

Query Error: arguments are located on different GPUs ... #60

Open phucty opened 4 years ago

phucty commented 4 years ago

Could you please help me solve it? I got an error when running the example query [1], the previous step is OK as the tutorial.

[1] curl "http://localhost:8000/travel/_search?pretty&q=passage:vegas&size=2" {"doc":"Unspecified run-time error.","msg":"('arguments are located on different GPUs at /pytorch/aten/src/THC/generic/THCTensorIndex.cu:397',)","type":"RuntimeError"}

pertschuk commented 4 years ago

are you running on multi-gpu environment?