diff --git a/torchrl/__init__.py b/torchrl/__init__.py index a7372f9ad69..10cbdda7600 100644 --- a/torchrl/__init__.py +++ b/torchrl/__init__.py @@ -29,10 +29,6 @@ set_lazy_legacy(False).set() -if torch.cuda.device_count() > 1: - n = torch.cuda.device_count() - 1 - os.environ["MUJOCO_EGL_DEVICE_ID"] = str(1 + (os.getpid() % n)) - from ._extension import _init_extension # noqa: E402 __version__ = None # type: ignore