pytorch-cpp icon indicating copy to clipboard operation
pytorch-cpp copied to clipboard
verified publisher icon prabhuomkar

Add DockerfileCUDA102

Open pyun-ram opened this issue 3 years ago • 1 comments

Description of the change

Description here

Type Of Change

  • [ ] Bug Fix (non-breaking change that fixes an issue)
  • [x] New Feature
  • [ ] New PyTorch tutorial
  • [ ] Breaking Change (cmake changes, fix or feature that would cause existing functionality to not work as expected)

Related Issues

Fix #100

Development & Code Review

  • [x] cpplint rules passes locally (run cmake -P cpplint.cmake)
  • [ ] CI is passing
  • [ ] Changes have been reviewed by at least one of the maintainers

pyun-ram avatar Oct 08 '22 14:10 pyun-ram

IMHO it would make more sense to extend the the current Dockerfile to include the functionality of building for/running on GPU. Then we could keep the extra features of our docker_entrypoint.sh.

mfl28 avatar Jan 03 '23 12:01 mfl28