Bug using colab gpu

[Tom_Bromley]

Hi @Neel_Gandhi and welcome to the forum!

Please include a minimal working example of the code you are using so we can best help.

However, from a quick look at what you’ve posted, it looks like you are using the Torch interface and trying to run on GPUs. Running on GPUs is not fully supported in PennyLane, you can check out previous posts, for example:

Pennylane and Pytorch running on GPU

I have two separate codes – one with classical machine learning (nothing to do with pennylane) and one with quantum ml (below). The only difference between the two is that the qml has additional pennylane code in the DQN class. Running the classical ml code runs without a problem on the GPU but when I run the qml code I get an error. Here is part of the code: import gym import math import random import numpy as np import matplotlib import matplotlib.pyplot as plt from collections import namedtu…

For now, I’d recommend running on CPU.

Thanks!