search

dqwang122 / HeterSumGraph

Code for ACL2020 paper "Heterogeneous Graph Neural Networks for Extractive Document Summarization"
245 stars 52 forks source link

Cuda version is not compatible with GPU while running the code in T4 GPU runtime of colab. #44

Open HanzalahI opened 8 months ago

HanzalahI commented 8 months ago

CODE: !python HeterSumGraph/evaluation.py --cuda --gpu 1 --data_dir /content/dataset/cnndm --cache_dir /content/graphfile/cache/CNNDM --embedding_path /content/glove.6B.300d.txt --model HSG --save_root /content/models/ --log_root /content/logs -m 3 --test_model multi --use_pyrouge ERROR: File "/content/HeterSumGraph/evaluation.py", line 239, in main() File "/content/HeterSumGraph/evaluation.py", line 227, in main model.to(torch.device("cuda:0")) File "/usr/local/lib/python3.10/dist-packages/torch/nn/modules/module.py", line 1160, in to return self._apply(convert) File "/usr/local/lib/python3.10/dist-packages/torch/nn/modules/module.py", line 810, in apply module._apply(fn) File "/usr/local/lib/python3.10/dist-packages/torch/nn/modules/module.py", line 833, in _apply param_applied = fn(param) File "/usr/local/lib/python3.10/dist-packages/torch/nn/modules/module.py", line 1158, in convert return t.to(device, dtype if t.is_floating_point() or t.is_complex() else None, non_blocking) File "/usr/local/lib/python3.10/dist-packages/torch/cuda/_init.py", line 298, in _lazy_init torch._C._cuda_init() RuntimeError: No CUDA GPUs are available

TuanAnh30 commented 4 months ago

hi, how can u have cache and glove file? thank u