Add instructions on how to select a GPU runtime in the tutorial notebook.
"\n",
"In this tutorial colab, we'll take a more in-depth look at Acme components by not\n",
"using the `D4PGBuilder` nor the `run_experiment` function and building the\n",
- "agent's components and connecting them manually."
+ "agent's components and connecting them manually.\n",
+ "\n",
+ "This colab is best run with a GPU runtime, and in particular, the last cell will\n",
+ "not run without it. (Please bear with us while we remove this requirement.) In\n",
+ "order to select a GPU runtime, click on `Runtime \u003e Change runtime type` and then\n",
+ "select `GPU` from the `Hardware type` dropdown menu."
]
},
{
},
"outputs": [],
"source": [
+ "# This is required for now as this cell uses GPU rendering. We are working on\n",
+ "# removing this requirement. See the instructions in the title cell to select a\n",
+ "# runtime with a GPU.\n",
+ "%env MUJOCO_GL=egl\n",
+ "\n",
"# Make the environment render frames and produce videos of episodes.\n",
"eval_environment = wrappers.MujocoVideoWrapper(environment, record_every=1)\n",
"\n",
