light-task-scheduler
light-task-scheduler copied to clipboard
有这个需求, 能不能帮我写个任务调度,并且能分配GPU/CUp 资源的
有这个需求, 能不能帮我写个任务调度,并且能分配GPU/CUp 资源的
要分配GPU/CPU应该要用 docker 吧,应该要实现一个 docker 执行器,类似框架里面的 https://github.com/memory-overflow/light-task-scheduler/blob/main/actuator/function_actuator.go 函数执行器。
目前还不涉及多个主机, 业务场景就是在一个docker 容器内, 如果有4个CPU, 8个GPU, 这种的资源分配
我最新比较忙,你看你能不能自己实现这个 docker 执行器。
执行go mod tidy 直接报错: github.com/memory-overflow/light-task-scheduler/actuator: module github.com/memory-overflow/light-task-scheduler@latest found (v1.0.2), but does not contain package github.com/memory-overflow/light-task-scheduler/actuator
go get -u github.com/memory-overflow/light-task-scheduler 就可以了
我实现了一个 docker actuator,https://github.com/memory-overflow/light-task-scheduler/blob/dev/docker_actuator/example/docker_example/main.go 参考这个 example。目前看起来 docker 可以分配内存、cpu,但是不好分配 gpu。