Skip to content
This repository was archived by the owner on Jan 7, 2025. It is now read-only.
This repository was archived by the owner on Jan 7, 2025. It is now read-only.

Torch GPU selection #130

Description

@lukeyeager

@Sravan2j

I just merged master into torch. As I did, I noticed something weird about the GPU assignment.

On this line, I copied your logic from before about adding 1 to the GPU id. Now when I create one torch job on GPU #0 the argument passed to torch is --devid=1, and when I assign another job to GPU #1 the argument is --devid=2. So far so good.

But then I look at nvidia-smi, and both jobs are running on GPU #1 (the second one). What's up with that?

Activity

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Metadata

Metadata

Assignees

No one assigned

    Labels

    Type

    No type

    Projects

    No projects

      Milestone

      No milestone

      Relationships

      None yet

      Development

      No branches or pull requests

      Issue actions