-
-
Notifications
You must be signed in to change notification settings - Fork 1.4k
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
base: main
Are you sure you want to change the base?
Added Dockerfile #738
Conversation
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 |
Oh thanks a lot - very interesting as well - great work! |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
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? |
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.