where to set cuda device? #261

Closed
opened 2021-08-31 05:12:52 +02:00 by sunian6183 · 2 comments
sunian6183 commented 2021-08-31 05:12:52 +02:00 (Migrated from github.com)

where to set cude device for choosing which one GPU to run ?

where to set cude device for choosing which one GPU to run ?
mive93 commented 2021-09-01 19:10:03 +02:00 (Migrated from github.com)

Do you mean when you have more than one GPU?
It's the same as other programs

export CUDA_VISIBLE_DEVICES=1

or

CUDA_VISIBLE_DEVICES=1 ./demo <all the param>
Do you mean when you have more than one GPU? It's the same as other programs ``` export CUDA_VISIBLE_DEVICES=1 ``` or ``` CUDA_VISIBLE_DEVICES=1 ./demo <all the param> ```
mive93 commented 2021-11-25 11:12:16 +01:00 (Migrated from github.com)

Closing for inactivity

Closing for inactivity
This repo is archived. You cannot comment on issues.
1 Participants
Due Date
No due date set.
Dependencies

No dependencies set.

Reference: mmr/tkDNN#261