RuntimeError: Device index must not be negative

[YeJinPaark]

08/18/2023 05:15:37 - INFO - /mnt/ssd_mnt/pyj/ASP/util/runner.py - Log file path: /mnt/ssd_mnt/pyj/ASP/data/ontonotes_coref/t5_base/log_Aug18_05-15-37.txt
08/18/2023 05:15:37 - INFO - util.tensorize_coref - Loaded tensorized examples from cache: /mnt/ssd_mnt/pyj/ASP/data/ontonotes_coref/cached.tensors.t5-small.2048.bin
/opt/conda/envs/asp/lib/python3.8/site-packages/torch/cuda/init.py:546: UserWarning: Can't initialize NVML
warnings.warn("Can't initialize NVML")
Traceback (most recent call last):
File "run_coref.py", line 121, in
model, _ = runner.initialize_model()
File "/mnt/ssd_mnt/pyj/ASP/util/runner.py", line 78, in initialize_model
model = self.model_class_fn(self.config, self.device)
File "/mnt/ssd_mnt/pyj/ASP/models/model_coref.py", line 49, in init
self.model = T5Coref.from_pretrained(
File "/opt/conda/envs/asp/lib/python3.8/site-packages/transformers/modeling_utils.py", line 2228, in from_pretrained
model = cls(config, *model_args, **model_kwargs)
File "/mnt/ssd_mnt/pyj/ASP/models/t5_coref.py", line 62, in init
self.action_head = util.make_ffnn(
File "/mnt/ssd_mnt/pyj/ASP/util/func.py", line 402, in make_ffnn
ffnn = [make_linear(feat_size, hidden_size[0], std=std), ACT2FN[activation], dropout]
File "/mnt/ssd_mnt/pyj/ASP/util/func.py", line 385, in make_linear
linear = nn.Linear(
File "/opt/conda/envs/asp/lib/python3.8/site-packages/torch/nn/modules/linear.py", line 96, in init
self.weight = Parameter(torch.empty((out_features, in_features), **factory_kwargs))
RuntimeError: Device index must not be negative

How can I fix it??

[7jun-ui]

It needs GPU