PyTorch 0.3.1 and CUDA 10.1 were used during development?

yuzhimanhua / Multi-BioNER

Cross-type Biomedical Named Entity Recognition with Deep Multi-task Learning (Bioinformatics'19)

https://arxiv.org/abs/1801.09851

Apache License 2.0

131 stars 28 forks source link

PyTorch 0.3.1 and CUDA 10.1 were used during development? #23

Closed Maxwell-ur closed 2 years ago

Maxwell-ur commented 2 years ago

Hi: When I configure the environment for your code. I find that torch 0.3.1 is very old version. Its corresponding CUDA version is 8 or 9, lower than 10.1. PyTorch 0.3.1 and CUDA 10.1 doesn't seem to match?

yuzhimanhua commented 2 years ago

Thank you for reporting this issue!

My bad. We did several rounds of modifications on README, and there are some inconsistencies. Please use torch 0.3.1 and CUDA 9.0.

Maxwell-ur commented 2 years ago

ok, thanks.