Hi, the file is installed by conda but run the test script report error. detail error information as follows:
Loading 4 IgFold models...
Using device: cuda:0
Loading /home/lyw/.local/lib/python3.8/site-packages/igfold/trained_models/IgFold/igfold_1.ckpt...
/home/lyw/.local/lib/python3.8/site-packages/torch/cuda/__init__.py:132: UserWarning:
Found GPU0 NVIDIA GeForce GT 720 which is of cuda capability 3.5.
PyTorch no longer supports this GPU because it is too old.
The minimum cuda capability supported by this library is 3.7.
warnings.warn(old_gpu_warn % (d, name, major, minor, min_arch // 10, min_arch % 10))
Loading /home/lyw/.local/lib/python3.8/site-packages/igfold/trained_models/IgFold/igfold_2.ckpt...
Loading /home/lyw/.local/lib/python3.8/site-packages/igfold/trained_models/IgFold/igfold_3.ckpt...
Loading /home/lyw/.local/lib/python3.8/site-packages/igfold/trained_models/IgFold/igfold_5.ckpt...
Successfully loaded 4 IgFold models.
Loaded AntiBERTy model.
Traceback (most recent call last):
File "test.py", line 9, in <module>
igfold.fold(
File "/home/lyw/.local/lib/python3.8/site-packages/igfold/IgFoldRunner.py", line 106, in fold
model_out = fold(
File "/home/lyw/.local/lib/python3.8/site-packages/igfold/utils/folding.py", line 182, in fold
embeddings, attentions = antiberty.embed(
File "/home/lyw/.local/lib/python3.8/site-packages/antiberty/AntiBERTyRunner.py", line 81, in embed
embeddings[i] = embeddings[i][:, a == 1]
RuntimeError: indices should be either on cpu or on the same device as the indexed tensor (cpu)
Hi, the file is installed by conda but run the test script report error. detail error information as follows:
Loading 4 IgFold models... Using device: cuda:0 Loading /home/lyw/.local/lib/python3.8/site-packages/igfold/trained_models/IgFold/igfold_1.ckpt... /home/lyw/.local/lib/python3.8/site-packages/torch/cuda/__init__.py:132: UserWarning: Found GPU0 NVIDIA GeForce GT 720 which is of cuda capability 3.5. PyTorch no longer supports this GPU because it is too old. The minimum cuda capability supported by this library is 3.7. warnings.warn(old_gpu_warn % (d, name, major, minor, min_arch // 10, min_arch % 10)) Loading /home/lyw/.local/lib/python3.8/site-packages/igfold/trained_models/IgFold/igfold_2.ckpt... Loading /home/lyw/.local/lib/python3.8/site-packages/igfold/trained_models/IgFold/igfold_3.ckpt... Loading /home/lyw/.local/lib/python3.8/site-packages/igfold/trained_models/IgFold/igfold_5.ckpt... Successfully loaded 4 IgFold models. Loaded AntiBERTy model. Traceback (most recent call last): File "test.py", line 9, in <module> igfold.fold( File "/home/lyw/.local/lib/python3.8/site-packages/igfold/IgFoldRunner.py", line 106, in fold model_out = fold( File "/home/lyw/.local/lib/python3.8/site-packages/igfold/utils/folding.py", line 182, in fold embeddings, attentions = antiberty.embed( File "/home/lyw/.local/lib/python3.8/site-packages/antiberty/AntiBERTyRunner.py", line 81, in embed embeddings[i] = embeddings[i][:, a == 1] RuntimeError: indices should be either on cpu or on the same device as the indexed tensor (cpu)