Update mario_rl_tutorial.py

ahoblitz · web-flow · commit 28a9d4b45404 · 2023-11-03T14:22:16.000-04:00
This should fix pytorch#2627
diff --git a/intermediate_source/mario_rl_tutorial.py b/intermediate_source/mario_rl_tutorial.py
@@ -32,6 +32,8 @@
 #
 #      %%bash
 #      pip install gym-super-mario-bros==7.4.0
+#      pip install tensordict==0.2.0
+#      pip install torchrl==0.2.0
 
 import torch
 from torch import nn

Original file line number	Diff line number	Diff line change
`@@ -32,6 +32,8 @@`
`32`	`32`	`#`
`33`	`33`	`# %%bash`
`34`	`34`	`# pip install gym-super-mario-bros==7.4.0`
	`35`	`+# pip install tensordict==0.2.0`
	`36`	`+# pip install torchrl==0.2.0`
`35`	`37`
`36`	`38`	`import torch`
`37`	`39`	`from torch import nn`