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

[Not an issue] Run sbb_pixelwise_segmentation in a docker image #19

Open
johnlockejrr opened this issue Oct 22, 2024 · 2 comments
Open

Comments

@johnlockejrr
Copy link

johnlockejrr commented Oct 22, 2024

If anybody is like me on CUDA 12.7 and can't train a model on keras v2 because is on keras v3 you can do it via a docker image:

docker pull nvidia/cuda:11.8.0-cudnn8-devel-ubuntu22.04
docker run --rm --name ubuntu22.04-nvidia-cuda11.8 --ipc=host -v E:/Docker/mount/sbb_train:/home/incognito/sbb_pixelwise_segmentation/train:rw -it --gpus=all nvidia/cuda:11.8.0-cudnn8-devel-ubuntu22.04 bash
  • I added a user incognito where I cloned the sbb_pixelwise_segmentation repo, made an virualenv, installed all the dependencies, then all the training I did in /home/incognito/sbb_pixelwise_segmentation/train so will be saved when shutting down the container.

Here is the container I used:
https://hub.docker.com/layers/nvidia/cuda/11.8.0-cudnn8-devel-ubuntu22.04/images/sha256-5079d0d0f36cc63050a0f5c010d769c68b8e959c2d2a45f8ec44dd7e5c1bf7f9

@cneud
Copy link
Member

cneud commented Oct 24, 2024

Hi @johnlockejrr - first of all, thanks for your interest and contributions in #17.

Our plan is to integrate the codebase from the unifying-training-models branch as train into the main codebase of Eynollah - e.g. to have the same (and more up-to-date) versions for dependencies and to reuse code. Once this is done (in a few weeks), there will automatically be also container builds available via docker pull ghcr.io/qurator-spk/eynollah:latest. Meanwhile, thanks for publishing a CUDA 12 compatible image.

@johnlockejrr
Copy link
Author

Happy to hear that! I love your project, it is really good and has even much potential.

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

No branches or pull requests

2 participants