Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Fix docker python version #2845

Merged
merged 2 commits into from
Feb 14, 2024
Merged

Conversation

NikolaBorisov
Copy link
Contributor

@NikolaBorisov NikolaBorisov commented Feb 13, 2024

Docker images were broken because we build the extensions with python3.8 but then we try to run them with other version of python.

This results in exception error

  File "/workspace/vllm/model_executor/layers/quantization/awq.py", line 6, in <module>
    from vllm._C import ops
ModuleNotFoundError: No module named 'vllm._C'

I think if we want to make sure vllm works in older versions of python we should do something else.

To reproduce
docker build .
docker run ...
It crashes with the above exception. I think as part of the CI there should be something that checks if the docker build containers work.

@NikolaBorisov
Copy link
Contributor Author

#2716 caused the issue

@NikolaBorisov
Copy link
Contributor Author

@simon-mo @WoosukKwon can someone take a look?

@NikolaBorisov
Copy link
Contributor Author

#2811 Also broke the docker images. This PR should fix it as well

@Pernekhan
Copy link
Contributor

LGTM!

@simon-mo
Copy link
Collaborator

Ah great catch! can you help refactor the Dockerfile so the test target uses 3.8 and everything else uses latest stable?

@simon-mo
Copy link
Collaborator

I guess this will close #2773 as well?

@NikolaBorisov
Copy link
Contributor Author

Yes this will fix #2773. I don't think we can have the python 3.8 just in the test target, because the issue was that c code was build with 3.8 but then it could not be imported in 3.10. I think to make sure 3.8 works we could build a different docker file or target that starts with 3.8 python and install vllm from pip? You will not catch it fast.

Docker images were broken because we build the extensions with
python3.8 but then we try to run them with other version of python.

This results in exception error

  File "/workspace/vllm/model_executor/layers/quantization/awq.py", line 6, in <module>
    from vllm._C import ops
ModuleNotFoundError: No module named 'vllm._C'

I think if we want to make sure vllm works in older versions of python
we should do something else.

Also fix for the cupy dependency
The docker build was not working with the requirement file
@simon-mo simon-mo merged commit 87069cc into vllm-project:main Feb 14, 2024
19 checks passed
xjpang pushed a commit to xjpang/vllm that referenced this pull request Feb 20, 2024
xjpang pushed a commit to xjpang/vllm that referenced this pull request Feb 22, 2024
xjpang pushed a commit to xjpang/vllm that referenced this pull request Mar 4, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

None yet

3 participants