fix: dont force nvidia runtime for users using Docker native GPU support
This fixes a bug for users who wouldn't have the nvidia runtime installed and instead use Docker's native GPU support.
@@ -31,7 +31,6 @@ const (
const (
// Container runtimes.
- nvidiaContainerRuntime = "nvidia"
runc = "runc"
)
@@ -134,8 +133,6 @@ func (t *TaskSpec) ToDockerSpec() container.Spec {
switch deviceType {
case device.CPU, device.ZeroSlot:
containerRuntime = runc
- case device.GPU:
- containerRuntime = nvidiaContainerRuntime
default:
panic(fmt.Sprintf("bad device type: %s", deviceType))
}
