Fix torchrl and tensordict requirements (#2986)

Vincent Moens · web-flow · commit fda818463b6d · 2024-08-01T02:15:16.000+01:00
diff --git a/.ci/docker/requirements.txt b/.ci/docker/requirements.txt
@@ -28,9 +28,8 @@ tensorboard
 jinja2==3.1.3
 pytorch-lightning
 torchx
-# TODO: use stable 0.5 when released
--e git+https://github.com/pytorch/rl.git#egg=torchrl
--e git+https://github.com/pytorch/tensordict.git#egg=tensordict
+torchrl==0.5.0
+tensordict==0.5.0
 ax-platform>==0.4.0
 nbformat>==5.9.2
 datasets
diff --git a/intermediate_source/pinmem_nonblock.py b/intermediate_source/pinmem_nonblock.py
@@ -556,7 +556,7 @@ def pin_copy_to_device_nonblocking(*tensors):
 # CUDA-enabled device.
 #
 # In summary, copying data from CPU to GPU is safe when using ``non_blocking=True``, but for any other direction,
-# ``non_blocking=True`` can still be used but the user must make sure that a device synchronization is executed after
+# ``non_blocking=True`` can still be used but the user must make sure that a device synchronization is executed before
 # the data is accessed.
 #
 # Practical recommendations

Original file line number	Diff line number	Diff line change
`@@ -556,7 +556,7 @@ def pin_copy_to_device_nonblocking(*tensors):`
`556`	`556`	`# CUDA-enabled device.`
`557`	`557`	`#`
`558`	`558`	# In summary, copying data from CPU to GPU is safe when using ``non_blocking=True``, but for any other direction,
`559`		-# ``non_blocking=True`` can still be used but the user must make sure that a device synchronization is executed after
	`559`	+# ``non_blocking=True`` can still be used but the user must make sure that a device synchronization is executed before
`560`	`560`	`# the data is accessed.`
`561`	`561`	`#`
`562`	`562`	`# Practical recommendations`