ModuleNotFoundError: No module named 'deepspeed'

[Ishihara-Masabumi]

A part of my environment for DisCo_Demo.ipynb is as follows:

!pip list
...
...

deepspeed                        0.8.2
...
...

But the following error occurred.

[2023-10-24 06:36:41 <wutils_ldm.py:150> <module>] <utils.py>: Deep Learning Utils @ Chenfei Wu
failed to load wandb
2023-10-24 06:36:43,418.418 2075:common.py:1785 setup_yaml(): python 3 env
failed to load wandb
---------------------------------------------------------------------------
ModuleNotFoundError                       Traceback (most recent call last)
[<ipython-input-9-7aa731641df4>](https://localhost:8080/#) in <cell line: 6>()
      4 
      5 from utils.wutils_ldm import *
----> 6 from agent import Agent_LDM, WarmupLinearLR, WarmupLinearConstantLR
      7 import torch
      8 from config import BasicArgs

[/content/DisCo/agent.py](https://localhost:8080/#) in <module>
      7 from utils.misc import humanbytes
      8 from utils.deepspeed import get_deepspeed_config, fp32_to_fp16
----> 9 import deepspeed
     10 from torch import nn
     11 import torch.nn.functional as F

ModuleNotFoundError: No module named 'deepspeed'

Please let me know how to fix it.