commit | 503d8981374a746cc9be4fa30ac662f46ce7673e | [log] [tgz] |
---|---|---|
author | Charles <charlie850220@gmail.com> | Thu Apr 15 11:18:36 2021 +0800 |
committer | Liu Xun <liuxun@apache.org> | Thu Apr 22 09:36:41 2021 +0000 |
tree | e44d1a77d3fa96a9ce076236701201246e920051 | |
parent | b0ea9583d24418653ef341681d68fedb40219dc6 [diff] |
SUBMARINE-697. Provide an environment whose image includes CUDA ### What is this PR for? Add an notebook environment which include CUDA v10.2 ### What type of PR is it? [Improvement] ### Todos * [ ] - Task ### What is the Jira issue? [SUBMARINE-697](https://issues.apache.org/jira/projects/SUBMARINE/issues/SUBMARINE-697?filter=myopenissues) ### How should this be tested? [Travic CI](https://travis-ci.org/github/charlie0220/submarine/builds/764863993) ### Screenshots (if appropriate) ![cuda-env](https://user-images.githubusercontent.com/50860453/112956379-fb19bf80-9172-11eb-8916-a6796c5e8cfb.png) ### Questions: * Does the licenses files need update? No * Is there breaking changes for older versions? No * Does this needs documentation? No Author: Charles <charlie850220@gmail.com> Signed-off-by: Liu Xun <liuxun@apache.org> Closes #546 from charlie0220/SUBMARINE-697 and squashes the following commits: 454e8b8 [Charles] Delete replicated word in database. b33f883 [Charles] Change image name from "cuda" to "gpu" 29a00f6 [Charles] SUBMARINE-697. Provide an environment whose image includes CUDA
Apache Submarine (Submarine for short) is an End-to-End Machine Learning Platform to allow data scientists to create end-to-end machine learning workflows. On Submarine, data scientists can finish each stage in the ML model lifecycle, including data exploration, data pipeline creation, model training, serving, and monitoring.
Some open-source and commercial projects are trying to build an end-to-end ML platform. What's the vision of Submarine?
Theodore Levitt once said:
“People don’t want to buy a quarter-inch drill. They want a quarter-inch hole.”
experiment
on prem or cloud via easy-to-use UI/API/SDK.experiment
and dependencies of environment
As mentioned above, Submarine attempts to provide Data-Scientist-friendly UI to make data scientists have a good user experience. Here're some examples.
# New a submarine client of the submarine server submarine_client = submarine.ExperimentClient(host='http://localhost:8080') # The experiment's environment, could be Docker image or Conda environment based environment = EnvironmentSpec(image='apache/submarine:tf-dist-mnist-test-1.0') # Specify the experiment's name, framework it's using, namespace it will run in, # the entry point. It can also accept environment variables. etc. # For PyTorch job, the framework should be 'Pytorch'. experiment_meta = ExperimentMeta(name='mnist-dist', namespace='default', framework='Tensorflow', cmd='python /var/tf_dist_mnist/dist_mnist.py --train_steps=100') # 1 PS task of 2 cpu, 1GB ps_spec = ExperimentTaskSpec(resources='cpu=2,memory=1024M', replicas=1) # 1 Worker task worker_spec = ExperimentTaskSpec(resources='cpu=2,memory=1024M', replicas=1) # Wrap up the meta, environment and task specs into an experiment. # For PyTorch job, the specs would be "Master" and "Worker". experiment_spec = ExperimentSpec(meta=experiment_meta, environment=environment, spec={'Ps':ps_spec, 'Worker': worker_spec}) # Submit the experiment to submarine server experiment = submarine_client.create_experiment(experiment_spec=experiment_spec) # Get the experiment ID id = experiment['experimentId']
submarine_client.get_experiment(id)
submarine_client.wait_for_finish(id)
submarine_client.get_log(id)
submarine_client.list_experiments(status='running')
For a quick-start, see Submarine On K8s
(Available on 0.6.0, see Roadmap)
If you want to know more about Submarine's architecture, components, requirements and design doc, they can be found on Architecture-and-requirement
Detailed design documentation, implementation notes can be found at: Implementation notes
Read the Apache Submarine Community Guide
How to contribute Contributing Guide
Issue Tracking: https://issues.apache.org/jira/projects/SUBMARINE
What to know more about what's coming for Submarine? Please check the roadmap out: https://cwiki.apache.org/confluence/display/SUBMARINE/Roadmap
The Apache Submarine project is licensed under the Apache 2.0 License. See the LICENSE file for details.