Closed ashok-arora closed 6 months ago
Hello, what is your tensorflow probability version? and tensorflow version?
SBX was successfully tested with tensorflow-probability==0.20.1
, tensorflow-probability==0.23.0
and without tensorflow installed.
I have tensorflow-probability==0.21.1
installed.
I have tensorflow-probability==0.21.1 installed.
and tensorflow? if you have tensorflow installed, try uninstalling it.
You can also upgrade tf probability.
Anyway, it sounds more like a tf probablity issue than a sbx one, I guess if you import tf proba and try to use it, you would have the same error?
In that case, I would open an issue their repo.
tensorflow version is 2.13.1
.
I uninstalled tensorflow and now got a different error:
Traceback (most recent call last):
File "check_sbx.py", line 2, in <module>
import stable_baselines3 as sb3
File "/home/user/miniconda3/envs/decision-transformer-gym/lib/python3.8/site-packages/stable_baselines3/__init__.py", line 3, in <module>
from stable_baselines3.a2c import A2C
File "/home/user/miniconda3/envs/decision-transformer-gym/lib/python3.8/site-packages/stable_baselines3/a2c/__init__.py", line 1, in <module>
from stable_baselines3.a2c.a2c import A2C
File "/home/user/miniconda3/envs/decision-transformer-gym/lib/python3.8/site-packages/stable_baselines3/a2c/a2c.py", line 7, in <module>
from stable_baselines3.common.buffers import RolloutBuffer
File "/home/user/miniconda3/envs/decision-transformer-gym/lib/python3.8/site-packages/stable_baselines3/common/buffers.py", line 16, in <module>
from stable_baselines3.common.utils import get_device
File "/home/user/miniconda3/envs/decision-transformer-gym/lib/python3.8/site-packages/stable_baselines3/common/utils.py", line 20, in <module>
from torch.utils.tensorboard import SummaryWriter
File "/home/user/miniconda3/envs/decision-transformer-gym/lib/python3.8/site-packages/torch/utils/tensorboard/__init__.py", line 12, in <module>
from .writer import FileWriter, SummaryWriter # noqa: F401
File "/home/user/miniconda3/envs/decision-transformer-gym/lib/python3.8/site-packages/torch/utils/tensorboard/writer.py", line 18, in <module>
from ._embedding import get_embedding_info, make_mat, make_sprite, make_tsv, write_pbtxt
File "/home/user/miniconda3/envs/decision-transformer-gym/lib/python3.8/site-packages/torch/utils/tensorboard/_embedding.py", line 9, in <module>
_HAS_GFILE_JOIN = hasattr(tf.io.gfile, "join")
File "/home/user/miniconda3/envs/decision-transformer-gym/lib/python3.8/site-packages/tensorboard/lazy.py", line 65, in __getattr__
return getattr(load_once(self), attr_name)
AttributeError: module 'tensorflow' has no attribute 'io'
it seems that tf is still partially installed, i would recommend start from an empty virtual env, that should solve your issue.
I am using tensorboard for logging values so that might be the cause of partial tf, anything else I can do aside from empty venv?
tensorboard doesn't need tf normally.
incompatibility might be fixed by https://github.com/araffin/sbx/pull/28
commit: https://github.com/araffin/sbx/pull/28/commits/deabd7f1eece907e73299abca587e3df4fb1ddf6
🐛 Bug
To Reproduce
error:
Expected behavior
Should have started training.
System Info
Additional context
Add any other context about the problem here.
Checklist