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

add gpu & deep-learning support #152

Draft
wants to merge 8 commits into
base: master
Choose a base branch
from
Draft

Conversation

c0mm4nd
Copy link

@c0mm4nd c0mm4nd commented May 26, 2023

Motivation: support the simulation for the federated learning framework on blockchain
Current: no gpu support (told by Prof. Du on his lecture, SUSTech)
Approach: Docker's native gpu support https://docs.docker.com/compose/gpu-support/

TODO:

  • base docker images with pytorch/tensorflow
  • pytorch/tensorflow services (with server)
  • fed-learning (with blockchain) example

BTW: the deploy.resources field is also a good approach to set the limits and reservations, but I feel hard to write the template, which is holy inflexible. What's your attitude on rewriting the compilers and generating the docker-composer.yml with the PyYAML lib?

@kevin-w-du
Copy link
Member

This is very interesting. Once I settled down from my travel next week, let's have a chat using Wechat or other platform. I want to learn more about what you are trying to do, which looks very interesting. Please send an email to me ([email protected]), so I can send you my wechat id.

@kevin-w-du
Copy link
Member

So far, whether our emulator can support learning algorithms was asked by the audience in three of my presentations. This is definitely an important feature that we should add to the emulator. Thanks for your work.

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

Successfully merging this pull request may close these issues.

2 participants