Intel® Distribution of OpenVINO™ Toolkit
Community assistance about the Intel® Distribution of OpenVINO™ toolkit, OpenCV, and all aspects of computer vision-related on Intel® platforms.

Select device to run model optimizer on

Nemes__Adam
Beginner
441 Views

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
438 Views

Ok.

I found a solution:

Simply setting 

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

Greetings,


Glad to know that you had found a solution.

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



Sincerely,

Iffa


0 Kudos
Nemes__Adam
Beginner
419 Views

Yes, feel free to close it.

Thank you

0 Kudos
Iffa_Intel
Moderator
413 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


0 Kudos
Reply