Transfer learning error

PennyLane Help
41

Hey @Martin_Guo! Welcome to the forum :sunglasses:

You can train hybrid models on GPUs! However, using lightning gpu with other ML frameworks and their GPU capabilities isn’t something we currently support/ But, you indeed can train hybrid models all within, say, PyTorch’s GPU API :slight_smile:

There are a ton of other discussion posts about this (e.g., Pennylane and Pytorch running on GPU - #17 by Daniel_Wang).

Let me know if that helps!

42

This is exactly what I need! Awesone work your team did!
I already sorted it out.

1 Like
43

Awesome! So glad to hear that :grin: