dotnet / dotnet/machinelearning-modelbuilder

Proposal: add device ID in LocalGpuEnvironment

Open
#2,506 0 comments 0 reactions 0 assignees View on GitHub
Stale
Dominant language
Dockerfile
Stars
285
Forks
66
PR merge metrics
No merged PRs in 30d

Description

Device ID will be a useful feature in multi-gpu environment.

The original schema for LocalGpuEnvironment is
```json
{
"Type": "LocalGpu"
}
```

proposal schema
```json
{
"Type": "LocalGpu",
"DeviceId": "cuda:0" | "cuda:1"
}
```

Contributor guide

No contributing guide indexed for this repository

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.