版本问题

Pytorch 版本

此问题与 https://github.com/pytorch/rl/issues/689 相关。使用 PyTorch 版本 <2.0 并安装稳定包会导致未定义的符号错误。例如

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

如何重现

1. 创建一个 Colab Notebook（在 2022 年 11 月 24 日，Colab 环境默认安装了 Python 3.7 和 Pytorch 1.12）。

2. !pip install torchrl

3. import torchrl

在 Colab 中，你可以通过运行以下命令解决此问题

!pip3 install torch --extra-index-url https://download.pytorch.org/whl/cpu -U

在 !pip install torchrl 命令之前。这将安装最新的 pytorch。说明可以在此处找到。

解决方法

对此问题有两种解决方法

1. 在安装 torchrl 之前，安装/升级到最新的 pytorch 版本。

2. 如果你需要使用之前的 pytorch 版本：安装与你的 torch 发行版相关的 functorch 版本：例如 pip install functorch==0.2.0 并从源代码安装库 pip install git+https://github.com/pytorch/rl@<lib_version_here>。