RuntimeError: Not compiled with CUDA support,help please.my torch is 2.1.0 and my cuda is 12.0.
zhouyupeng973 opened this issue · 3 comments
File "train.py", line 25, in
model.train()
File "/home/libj/ZYP/1129-hccr/ZS-HCCR/networks/model.py", line 434, in train
notebook_launcher(self.train_loop, self.training_args, num_processes=self.opt.num_gpus)
File "/home/libj/.conda/envs/CONTROLNET2/lib/python3.8/site-packages/accelerate/launchers.py", line 196, in notebook_launcher
function(*args)
File "/home/libj/ZYP/1129-hccr/ZS-HCCR/networks/model.py", line 369, in train_loop
val_acc = self._validate(accelerator.device)
File "/home/libj/ZYP/1129-hccr/ZS-HCCR/networks/model.py", line 447, in _validate
template = self.graph_model(data)
File "/home/libj/.conda/envs/CONTROLNET2/lib/python3.8/site-packages/torch/nn/modules/module.py", line 1518, in _wrapped_call_impl
return self._call_impl(*args, **kwargs)
File "/home/libj/.conda/envs/CONTROLNET2/lib/python3.8/site-packages/torch/nn/modules/module.py", line 1527, in _call_impl
return forward_call(*args, **kwargs)
File "/home/libj/ZYP/1129-hccr/ZS-HCCR/gcn_networks/model.py", line 83, in forward
x = self.features(data)
File "/home/libj/.conda/envs/CONTROLNET2/lib/python3.8/site-packages/torch/nn/modules/module.py", line 1518, in _wrapped_call_impl
return self._call_impl(*args, **kwargs)
File "/home/libj/.conda/envs/CONTROLNET2/lib/python3.8/site-packages/torch/nn/modules/module.py", line 1527, in _call_impl
return forward_call(*args, **kwargs)
File "/home/libj/.conda/envs/CONTROLNET2/lib/python3.8/site-packages/torch/nn/modules/container.py", line 215, in forward
input = module(input)
File "/home/libj/.conda/envs/CONTROLNET2/lib/python3.8/site-packages/torch/nn/modules/module.py", line 1518, in _wrapped_call_impl
return self._call_impl(*args, **kwargs)
File "/home/libj/.conda/envs/CONTROLNET2/lib/python3.8/site-packages/torch/nn/modules/module.py", line 1527, in _call_impl
return forward_call(*args, **kwargs)
File "/home/libj/ZYP/1129-hccr/ZS-HCCR/gcn_networks/module.py", line 552, in forward
cluster = graclus(data.edge_index, weight, data.x.size(0))
File "/home/libj/.conda/envs/CONTROLNET2/lib/python3.8/site-packages/torch_geometric/nn/pool/graclus.py", line 35, in graclus
return graclus_cluster(edge_index[0], edge_index[1], weight, num_nodes)
File "/home/libj/.conda/envs/CONTROLNET2/lib/python3.8/site-packages/torch_cluster/graclus.py", line 61, in graclus_cluster
return torch.ops.torch_cluster.graclus(rowptr, col, weight)
File "/home/libj/.conda/envs/CONTROLNET2/lib/python3.8/site-packages/torch/_ops.py", line 692, in call
return self._op(*args, **kwargs or {})
RuntimeError: Not compiled with CUDA support
How did you install torch-cluster? You can also try to install via FORCE_CUDA=1 pip install ...
solved！thank you.