search

erprogs / GenConViT

Deepfake Video Detection Using Generative Convolutional Vision Transformer
GNU General Public License v3.0
51 stars 11 forks source link

What is the version of CUDA? #8

Open Charltin opened 8 months ago

Charltin commented 8 months ago

I use the cuda 12.2, but it doesn't work. Then I try to use the torch 2.2.1(for cuda 12.2).

python train.py --d sample_train_data --m vae -e 5 -t y

I got this "RuntimeError: GET was unable to find an engine to execute this computation"

erprogs commented 8 months ago

Hello @Charltin, I never encountered this error before. The model is trained on CUDA Version: 12.2. what is the output of torch.cuda.is_available()? please check if you have all the necessary libraries installed. also, can you run the pre-trained model for detection?

happy coding!