search

DeepChainBio / bio-transformers

bio-transformers is a wrapper on top of the ESM/Protbert model, trained on millions on proteins and used to predict embeddings.
https://bio-transformers.readthedocs.io/en/latest/getting_started/install.html
Apache License 2.0
143 stars 31 forks source link

CUDA RUNTIME ERROR #20

Closed idmjky closed 3 years ago

idmjky commented 3 years ago

Hi, Thanks for updating the multi-GPU instructions. I am wondering since there is no CUDA toolkit installed during the default installation, do I need to install this aside from the bio-transformers main packages? Also, when I try to run with multi-GPUs, I get this error RTX A6000 with CUDA capability sm_86 is not compatible with the current PyTorch installation. The current PyTorch install supports CUDA capabilities sm_37 sm_50 sm_60 sm_70. If you want to use the RTX A6000 GPU with PyTorch, please check the instructions at https://pytorch.org/get-started/locally/

delfosseaurelien commented 3 years ago

Hi, you still have this configuration?

Conda torch 1.8.1 NVIDIA-SMI Driver Version: 460.67 CUDA Version: 11.2 Conda cudatoolkit 11.0.221

idmjky commented 3 years ago

Yes, I found this seems to be a problem with A600 GPUs only. Others are fine. Thank you for your help.