I have installed trl using pip.

!pip show trl

and its showing packages info properly,

!pip show trl

Name: trl
Version: 0.7.10
Summary: Train transformer language models with reinforcement learning.
Home-page: 
Author: Leandro von Werra
Author-email: [email protected]
License: Apache 2.0
Location: /usr/local/lib/python3.10/dist-packages
Requires: accelerate, datasets, numpy, torch, transformers, tyro
Required-by: 

However, when I am importing in Notebook its showing error:

import trl

---------------------------------------------------------------------------
ModuleNotFoundError                       Traceback (most recent call last)
Cell In[9], line 1
----> 1 import trl

ModuleNotFoundError: No module named 'trl'

Why this discrepancy? How to resolve it?

I have installed trl using pip.

!pip show trl

and its showing packages info properly,

!pip show trl

Name: trl
Version: 0.7.10
Summary: Train transformer language models with reinforcement learning.
Home-page: https://github/huggingface/trl
Author: Leandro von Werra
Author-email: [email protected]
License: Apache 2.0
Location: /usr/local/lib/python3.10/dist-packages
Requires: accelerate, datasets, numpy, torch, transformers, tyro
Required-by: 

However, when I am importing in Notebook its showing error:

import trl

---------------------------------------------------------------------------
ModuleNotFoundError                       Traceback (most recent call last)
Cell In[9], line 1
----> 1 import trl

ModuleNotFoundError: No module named 'trl'

Why this discrepancy? How to resolve it?

• python
• jupyter-notebook
• pip
• package

1 Answer 1

The problem was solved when I installed from direct source using,

pip install git+https://github/huggingface/trl.git