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

Added Dockerfile #738

Open
wants to merge 3 commits into
base: main
Choose a base branch
from
Open

Added Dockerfile #738

wants to merge 3 commits into from

Conversation

dsingal0
Copy link

@dsingal0 dsingal0 commented Jul 6, 2024

Lots of people asking for a dockerfile for unsloth, so I added one. Things are very temperamental in the order pip packages are installed especially packaging and flash-attn. I don't see tests, but was checked by training tiny-llama.

@dsingal0
Copy link
Author

dsingal0 commented Jul 6, 2024

@dsingal0 dsingal0 mentioned this pull request Jul 7, 2024
@darkacorn
Copy link

worth makeing a builder for a few different cuda versions - depending on the arch .. not every card supports 12.1

@dsingal0
Copy link
Author

dsingal0 commented Jul 7, 2024

worth makeing a builder for a few different cuda versions - depending on the arch .. not every card supports 12.1

@darkacorn what cards don't support 12.1? I believe everything since Pascal does? It's probably worth adding one for 11.8 for people using older pytorch and pytorch dependent libraries

@danielhanchen
Copy link
Contributor

Oh thanks a lot - very interesting as well - great work!

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

I am getting thr following error when i try to import FastLanguageModel which i do in a script which is inside a docker container which is setup same as your Dockerfile
image

@GeraudBourdin
Copy link

By the way, you can do a specific repo and push images on docker.io this could be helpful for firs timers.

@dsingal0
Copy link
Author

By the way, you can do a specific repo and push images on docker.io this could be helpful for firs timers.

@GeraudBourdin what do you mean? I'd prefer not to own unsloth dockerfiles. @danielhanchen do you think you can merge so people can have a dockerfile for unsloth?

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.

5 participants