NameError: name 'base' is not defined, while running open AI gym in GOOGLE COLAB
I am going through the tutorial DQN reinforcement learning in Pytorch.org,https://pytorch.org/tutorials/intermediate/reinforcement_q_learning.html
But here when I am trying to render a screen and display using python display, I am getting name base not found. Can anyone help me here? If you want to clear any clarity about the question, I am here
Thanks in advance

Use the gpu on colab. If with that isn't enought execute this cell at the beginning
!apt install xvfb -y
!pip install pyvirtualdisplay
!pip install piglet

from pyvirtualdisplay import Display
display = Display(visible=0, size=(1400, 900))
display.start()

