search

capreolus-ir / capreolus

A toolkit for end-to-end neural ad hoc retrieval
https://capreolus.ai
Apache License 2.0
95 stars 32 forks source link

DeepTileBars failing on GPU with CUDA 10.1 #60

Closed andrewyates closed 4 years ago

andrewyates commented 4 years ago

The tests for DTB are now failing when a GPU with a recent CUDA version is used. Setting CUDA_VISIBLE_DEVICES='' causes the tests to pass.

andrewyates commented 4 years ago

This is a PyTorch/CUDA bug fixed in PyTorch 1.5.