Community
cancel
Showing results for 
Search instead for 
Did you mean: 
Nemes__Adam
Beginner
45 Views

Select device to run model optimizer on

I would like to run a model optimizer on a multi GPU system.

In case the first GPU is in use (not enough memory availble to load the model),

I would like the script to use the other GPU.

Nemes__Adam_0-1619609377972.png

How can i select the CUDA Device in such case?

 

0 Kudos
4 Replies
Nemes__Adam
Beginner
42 Views

Ok.

I found a solution:

Simply setting 

os.environ["CUDA_VISIBLE_DEVICES"] to the desired GPU does the job
Iffa_Intel
Moderator
30 Views

Greetings,


Glad to know that you had found a solution.

If you have no further inquiries, shall I close this thread?



Sincerely,

Iffa


Nemes__Adam
Beginner
23 Views

Yes, feel free to close it.

Thank you

Iffa_Intel
Moderator
17 Views

Greetings,


Intel will no longer monitor this thread since this issue has been resolved. If you need any additional information from Intel, please submit a new question.


Sincerely,

Iffa


Reply