No module named openai

I've installed openai on my laptop with pip install openai.

Have installed on my laptop and after installed on the same folder where my code file is. But when I try to run the code I get ImportError: No module named openai

This is the code from the file. Pretty simple:

import openaiopenai.api_key = API_KEYprompt = "Say this is a test"response = openai.Completion.create(    engine="text-davinci-001", prompt=prompt, max_tokens=6)print(response)

What am I doing wrong?