HeterSumGraph icon indicating copy to clipboard operation
HeterSumGraph copied to clipboard

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

Open HanzalahI opened this issue 1 year ago • 1 comments

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

HanzalahI avatar Feb 21 '24 06:02 HanzalahI