Разреженность факела с использованием версии CUDA среды вместо версии факела ⇐ Python

[Anonymous]

Я пытаюсь запустить скрипт Python в кластере, для которого требуется следующее:
torch==1.7.1

torch-sparse== 0.6.8
Проблема в том, что когда я устанавливаю оба из них, torch поставляется с версией cuda 10.2, а torch-sparse использует доступную версию в кластере, то есть 11, поэтому это дает следующая ошибка

Код: Выделить всё

Traceback (most recent call last):
File "/home/g.pocas/GraphGONet/scripts/GraphGONet.py", line 6, in 
from torch_geometric.data import DataLoader
File "/home/g.pocas/GraphGONet/venv/lib/python3.9/site-packages/torch_geometric/__init__.py", line 2, in 
import torch_geometric.nn
File "/home/g.pocas/GraphGONet/venv/lib/python3.9/site-packages/torch_geometric/nn/__init__.py", line 2, in 
from .data_parallel import DataParallel
File "/home/g.pocas/GraphGONet/venv/lib/python3.9/site-packages/torch_geometric/nn/data_parallel.py", line 5, in 
from torch_geometric.data import Batch
File "/home/g.pocas/GraphGONet/venv/lib/python3.9/site-packages/torch_geometric/data/__init__.py", line 1, in 
from .data import Data
File "/home/g.pocas/GraphGONet/venv/lib/python3.9/site-packages/torch_geometric/data/data.py", line 8, in 
from torch_sparse import coalesce, SparseTensor
File "/home/g.pocas/GraphGONet/venv/lib/python3.9/site-packages/torch_sparse/__init__.py", line 27, in 
raise RuntimeError(
RuntimeError: Detected that PyTorch and torch_sparse were compiled with different CUDA versions. PyTorch has CUDA version 10.2 and torch_sparse has CUDA version 11.2. Please reinstall the torch_sparse that matches your PyTorch install.

Я пробовал установить torch без cuda, и версия torch sparse должна быть совместима с torch 1.7.1, поскольку я установил ее непосредственно с предоставленного ими колеса, так что я действительно не уверен, что может быть причиной этого.

Подробнее здесь: https://stackoverflow.com/questions/792 ... hs-version