{"id":15436072,"url":"https://github.com/uetchy/dockerfile-machinelearning","last_synced_at":"2025-04-19T18:24:02.054Z","repository":{"id":35491173,"uuid":"39760642","full_name":"uetchy/dockerfile-machinelearning","owner":"uetchy","description":"🐳🤖 Dockerfile for ML researchers.","archived":false,"fork":false,"pushed_at":"2019-03-20T16:29:18.000Z","size":33,"stargazers_count":45,"open_issues_count":1,"forks_count":15,"subscribers_count":5,"default_branch":"master","last_synced_at":"2025-03-29T11:34:41.913Z","etag":null,"topics":["docker-image","dockerfile","machine-learning"],"latest_commit_sha":null,"homepage":"","language":"Dockerfile","has_issues":true,"has_wiki":null,"has_pages":null,"mirror_url":null,"source_name":null,"license":"mit","status":null,"scm":"git","pull_requests_enabled":true,"icon_url":"https://github.com/uetchy.png","metadata":{"files":{"readme":"README.md","changelog":null,"contributing":null,"funding":null,"license":"LICENSE","code_of_conduct":null,"threat_model":null,"audit":null,"citation":null,"codeowners":null,"security":null,"support":null}},"created_at":"2015-07-27T07:16:59.000Z","updated_at":"2021-10-26T23:05:57.000Z","dependencies_parsed_at":"2022-09-12T13:01:27.587Z","dependency_job_id":null,"html_url":"https://github.com/uetchy/dockerfile-machinelearning","commit_stats":null,"previous_names":[],"tags_count":0,"template":false,"template_full_name":null,"repository_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories/uetchy%2Fdockerfile-machinelearning","tags_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories/uetchy%2Fdockerfile-machinelearning/tags","releases_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories/uetchy%2Fdockerfile-machinelearning/releases","manifests_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories/uetchy%2Fdockerfile-machinelearning/manifests","owner_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/owners/uetchy","download_url":"https://codeload.github.com/uetchy/dockerfile-machinelearning/tar.gz/refs/heads/master","host":{"name":"GitHub","url":"https://github.com","kind":"github","repositories_count":249762276,"owners_count":21321905,"icon_url":"https://github.com/github.png","version":null,"created_at":"2022-05-30T11:31:42.601Z","updated_at":"2022-07-04T15:15:14.044Z","host_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub","repositories_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories","repository_names_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repository_names","owners_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/owners"}},"keywords":["docker-image","dockerfile","machine-learning"],"created_at":"2024-10-01T18:48:06.835Z","updated_at":"2025-04-19T18:24:02.023Z","avatar_url":"https://github.com/uetchy.png","language":"Dockerfile","funding_links":[],"categories":[],"sub_categories":[],"readme":"# Dockerfile for Machine Learning\n\nA Dockerfile for accelerated research process, consists of major machine learning libraries.\n\n- [DockerHub](https://registry.hub.docker.com/u/uetchy/ml/)\n\n## Features\n\n- Ubuntu 18.04\n- Python 3.6 (Miniconda 3)\n- GPU accelerated (CUDA 10.0, cuDNN 7)\n- NCCL, CNMeM, Apex (PyTorch only) activated\n- Jupyter and OpenCV 3.0 included\n- Additional packages (Tensorboard, Hyperdash, etc)\n\n## Available Deep Learning Frameworks\n\n- TensorFlow `uetchy/ml:tensorflow`\n- PyTorch `uetchy/ml:pytorch`\n- Chainer `uetchy/ml:chainer`\n- MXnet `uetchy/ml:mxnet`\n- XGBoost `uetchy/ml:xgboost`\n\n## Install\n\n### System Requirements\n\n- Docker\n- CUDA-enabled GPUs\n- CUDA Driver\n- CUDA Toolkit\n- nvidia-docker2 (https://github.com/NVIDIA/nvidia-docker)\n\n### Pull the docker image from [DockerHub](https://registry.hub.docker.com/u/uetchy/ml/)\n\n```bash\ndocker pull uetchy/ml:tensorflow\ndocker pull uetchy/ml:pytorch\ndocker pull uetchy/ml:chainer\n...\n```\n\n### Launch Jupyter Lab on current directory\n\n```bash\ndocker run --runtime=nvidia -v $PWD:/workspace -p 8888:8888 -it uetchy/ml:pytorch jupyter\nopen http://localhost:8888\n```\n\n### Open Python REPL\n\n```bash\ndocker run --runtime=nvidia --rm -it uetchy/ml:base python\n```\n\n### Run Bash Shell\n\n```bash\ndocker run --runtime=nvidia --rm -it uetchy/ml:tensorflow\n```\n\n## List of Docker images for Data Science\n\n- [TensorFlow](https://hub.docker.com/r/tensorflow/tensorflow)\n- [Caffe2](https://hub.docker.com/r/caffe2ai/caffe2)\n- [MXNet](https://hub.docker.com/u/mxnet)\n- [Caffe](https://github.com/BVLC/caffe/tree/master/docker)\n\n# Contribution\n\nPRs are accepted.\n\n## Contributors\n\n- Yasuaki Uechi\n- UpmostScarab\n- cyrusmvahid\n","project_url":"https://awesome.ecosyste.ms/api/v1/projects/github.com%2Fuetchy%2Fdockerfile-machinelearning","html_url":"https://awesome.ecosyste.ms/projects/github.com%2Fuetchy%2Fdockerfile-machinelearning","lists_url":"https://awesome.ecosyste.ms/api/v1/projects/github.com%2Fuetchy%2Fdockerfile-machinelearning/lists"}