Thanks to visit codestin.com
Credit goes to github.com

Skip to content

[BUG] PPO with torchrl tutorial broken on Colab #1628

@mikemykhaylov

Description

@mikemykhaylov

Describe the bug

When trying to run the tutorial on Colab, the following error is thrown when importing torchrl:

---------------------------------------------------------------------------
ImportError                               Traceback (most recent call last)
[<ipython-input-4-8f198eb0e7fb>](https://localhost:8080/#) in <cell line: 9>()
      7 from torch import nn
      8 
----> 9 from torchrl.collectors import SyncDataCollector
     10 from torchrl.data.replay_buffers import ReplayBuffer
     11 from torchrl.data.replay_buffers.samplers import SamplerWithoutReplacement

8 frames
[/usr/local/lib/python3.10/dist-packages/torchrl/data/replay_buffers/samplers.py](https://localhost:8080/#) in <module>
     11 import torch
     12 
---> 13 from torchrl._torchrl import (
     14     MinSegmentTreeFp32,
     15     MinSegmentTreeFp64,

ImportError: /usr/local/lib/python3.10/dist-packages/torchrl/_torchrl.so: undefined symbol: _ZN5torch8autograd13_wrap_outputsERKSt6vectorIN2at6TensorESaIS3_EERKSt13unordered_setIPN3c1010TensorImplESt4hashISB_ESt8equal_toISB_ESaISB_EESJ_NS9_8ArrayRefINS9_8optionalIS3_EEEERKSt10shared_ptrINS0_4NodeEESt8functionIFS5_S5_S5_EESJ_

---------------------------------------------------------------------------
NOTE: If your import is failing due to a missing package, you can
manually install dependencies using either !pip or !apt.

To view examples of installing some common dependencies, click the
"Open Examples" button below.
---------------------------------------------------------------------------

To Reproduce

  1. Import tutorial notebook on Colab
  2. Uncomment code installing the packages
  3. Run the code

Expected behavior

Error doesn't happen

Screenshots

If applicable, add screenshots to help explain your problem.

System info

Standard Google Colab En

Checklist

  • I have checked that there is no similar issue in the repo (required)
  • I have read the documentation (required)
  • I have provided a minimal working example to reproduce the bug (required)

Metadata

Metadata

Assignees

Labels

bugSomething isn't working

Type

No type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions