-
Notifications
You must be signed in to change notification settings - Fork 4
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
ENH: Added jupyter-pytorch-notebook Dockerfile #67
Conversation
Currently pytorch notebook test is failing because the image is close to max cap of GitHub actions space provisions |
31f292a
to
15742c8
Compare
mamba was causing problems with pytorch CUDA compatibility so used pip for installing the required packages instead
5fcecb7
to
3050395
Compare
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.
Looks good to me!
I don't know if it is possible or not but could you provide versions for transformers
, accelerate
etc, so that it is more reproducible?
fix-permissions "${CONDA_DIR}" && \ | ||
fix-permissions "/home/${NB_USER}" | ||
#switch back to jovyan to avoid accidental container runs as root | ||
USER ${NB_UID} |
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.
Not sure if it changes anything but does it matter if this variable has the curly braces, whereas on line 11 it is written without them USER $NB_UID
.
Otherwise it all looks good to me
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.
I don't think there is a difference between the two, but it keeps it consistent with how we switch back to user in our other dockerfiles
Installs
pytorch
,transformers
,accelarate
,bitsandbytes
using mamba