RuntimeError: nvrtc: error: invalid value for --gpu-architecture (-arch)

[Paul-YPH]

When I run
adata = STAGATE_pyG.train_STAGATE(adata, device = torch.device('cuda:0' if torch.cuda.is_available() else 'cpu'))
I got this problem.
I try to change the version of cuda, but it seems does't work.
I think maybe the version of pytorch led to the error, but could not find information in readme and requiement.txt.
Hoping for your reply.

[Paul-YPH]

Package Version

---

absl-py 1.0.0
anndata 0.8.0
anndata2ri 1.0.6
asttokens 2.0.5
astunparse 1.6.3
backcall 0.2.0
backports.zoneinfo 0.2.1
boto3 1.21.40
botocore 1.24.40
cachetools 4.2.4
certifi 2021.10.8
cffi 1.15.0
charset-normalizer 2.0.12
click 8.1.2
cycler 0.11.0
dataclasses 0.6
debugpy 1.5.1
decorator 5.1.1
deep-forest 0.1.5
dunamai 1.11.1
easydict 1.9
entrypoints 0.3
executing 0.8.3
flatbuffers 2.0
fonttools 4.32.0
future 0.18.2
gast 0.5.3
get_version 3.5.4
google-auth 1.35.0
google-auth-oauthlib 0.4.6
google-pasta 0.2.0
grpcio 1.44.0
h5py 3.6.0
idna 3.3
igraph 0.9.10
importlib-metadata 4.11.3
install 1.3.5
ipykernel 6.9.1
ipython 8.2.0
jedi 0.18.1
Jinja2 3.1.1
jmespath 1.0.0
joblib 1.1.0
jupyter-client 7.1.2
jupyter-core 4.9.2
keras 2.8.0
Keras-Preprocessing 1.1.2
kiwisolver 1.4.2
libclang 14.0.1
llvmlite 0.38.0
louvain 0.7.1
Markdown 3.3.6
MarkupSafe 2.1.1
matplotlib 3.5.1
matplotlib-inline 0.1.2
natsort 8.1.0
nest-asyncio 1.5.5
networkx 2.8
numba 0.55.1
numexpr 2.8.1
numpy 1.19.2
oauthlib 3.2.0
opt-einsum 3.3.0
packaging 21.3
pandas 1.1.3
parso 0.8.3
patsy 0.5.2
pexpect 4.8.0
pickleshare 0.7.5
Pillow 9.1.0
pip 21.2.4
prompt-toolkit 3.0.20
protobuf 3.20.0
ptyprocess 0.7.0
pure-eval 0.2.2
pyasn1 0.4.8
pyasn1-modules 0.2.8
pycparser 2.21
Pygments 2.11.2
pynndescent 0.5.6
pyparsing 3.0.8
python-dateutil 2.8.2
pytorch-transformers 1.2.0
pytz 2022.1
pytz-deprecation-shim 0.1.0.post0
pyzmq 22.3.0
regex 2022.3.15
requests 2.27.1
requests-oauthlib 1.3.1
rpy2 3.5.1
rsa 4.8
s3transfer 0.5.2
sacremoses 0.0.49
scanpy 1.8.1
scikit-learn 1.0.1
scikit-misc 0.1.4
scipy 1.5.2
seaborn 0.11.2
sentencepiece 0.1.96
setuptools 61.2.0
sinfo 0.3.4
six 1.16.0
stack-data 0.2.0
STAGATE 1.0.1
STAGATE-pyG 1.0.0
statsmodels 0.13.2
stdlib-list 0.8.0
tables 3.7.0
tensorboard 2.8.0
tensorboard-data-server 0.6.1
tensorboard-plugin-wit 1.8.1
tensorflow-io-gcs-filesystem 0.25.0
termcolor 1.1.0
texttable 1.6.4
tf-estimator-nightly 2.8.0.dev2021122109
threadpoolctl 3.1.0
torch 1.7.0+cu110
torch-geometric 2.0.4
torch-scatter 2.0.5
torch-sparse 0.6.8
torchaudio 0.7.0
torchvision 0.8.1+cu110
tornado 6.1
tqdm 4.50.2
traitlets 5.1.1
typing_extensions 4.1.1
tzdata 2022.1
tzlocal 4.2
umap-learn 0.5.3
Unidecode 1.3.0
urllib3 1.26.9
wcwidth 0.2.5
Werkzeug 2.1.1
wheel 0.37.1
wrapt 1.14.0
zipp 3.8.0