Skip to content
Projects
Groups
Snippets
Help
Loading...
Help
Support
Submit feedback
Contribute to GitLab
Sign in
Toggle navigation
P
power-node
Project
Project
Details
Activity
Releases
Cycle Analytics
Repository
Repository
Files
Commits
Branches
Tags
Contributors
Graph
Compare
Charts
Issues
0
Issues
0
List
Boards
Labels
Milestones
Merge Requests
0
Merge Requests
0
CI / CD
CI / CD
Pipelines
Jobs
Schedules
Charts
Wiki
Wiki
Snippets
Snippets
Members
Members
Collapse sidebar
Close sidebar
Activity
Graph
Charts
Create a new issue
Jobs
Commits
Issue Boards
Open sidebar
Odysseus
power-node
Commits
ddcbf1bd
Commit
ddcbf1bd
authored
Jun 04, 2024
by
duanjinfei
Browse files
Options
Browse Files
Download
Email Patches
Plain Diff
checkGpuUsage
parent
f30e0637
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
4 additions
and
2 deletions
+4
-2
docker.go
operate/docker.go
+4
-2
No files found.
operate/docker.go
View file @
ddcbf1bd
...
...
@@ -392,11 +392,13 @@ func (d *DockerOp) getContainerInfo(id string) (types.Container, error) {
func
(
d
*
DockerOp
)
checkGpuUsage
(
info
*
nodemanagerV2
.
HardwareInfo
,
modelInfo
*
models
.
ModelInfo
,
dockerCmd
*
models
.
DockerCmd
)
int32
{
envMap
:=
make
(
map
[
string
]
string
,
0
)
var
res
int32
=
999
gpu
:=
info
.
GPU
isMatch
:=
false
for
_
,
gpuInfo
:=
range
gpu
{
if
gpuInfo
.
MemFree
>
modelInfo
.
RunningMem
{
envMap
[
models
.
CudaEnv
]
=
strconv
.
FormatInt
(
int64
(
gpuInfo
.
Seq
),
10
)
res
=
gpuInfo
.
Seq
isMatch
=
true
break
}
...
...
@@ -418,7 +420,7 @@ func (d *DockerOp) checkGpuUsage(info *nodemanagerV2.HardwareInfo, modelInfo *mo
if
isMatch
{
//nm.ModelRunningBeforeMem[modelInfo.ImageName] = dockerCmd.RunningBeforeMem
gpuSeq
,
_
:=
strconv
.
ParseInt
(
dockerCmd
.
EnvMap
[
models
.
CudaEnv
],
10
,
32
)
re
turn
int32
(
gpuSeq
)
re
s
=
int32
(
gpuSeq
)
}
return
0
return
res
}
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment