What is the version of CUDA?

[Charltin]

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]

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!