RuntimeError: legacy constructor expects device type: cpubut device type: cuda was passed

[miderxi]

python main_inference.py --experiment_name dMaSIF_site_3layer_16dims_9A_0.7res_150sup_epoch85 --embedding_layer dMaSIF --emb_dims 16 --n_layers 3 --resolution 0.7 --radius 9 --single_pdb 1A0G_B_B --site True --device cuda:0 0%| | 0/1 [00:00<?, ?it/s] Traceback (most recent call last): File "main_inference.py", line 86, in pdb_ids=test_pdb_ids, File "/home/v/proj/dmasif/content/MaSIF_colab/data_iteration.py", line 286, in iterate P1_batch, P2_batch = process(args, protein_pair, net) File "/home/v/proj/dmasif/content/MaSIF_colab/data_iteration.py", line 126, in process net.preprocess_surface(P1) File "/home/v/proj/dmasif/content/MaSIF_colab/model.py", line 454, in preprocess_surface distance=self.args.distance, File "/home/v/proj/dmasif/content/MaSIF_colab/geometry_processing.py", line 266, in atoms_to_points_normals atomtypes=atomtypes, File "/home/v/proj/dmasif/content/MaSIF_colab/geometry_processing.py", line 164, in soft_distances [170, 110, 152, 155, 180, 190], device=x.device RuntimeError: legacy constructor expects device type: cpubut device type: cuda was passed