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

Select device to run model optimizer on

Nemes__Adam
Beginner
178 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
175 Views

Ok.

I found a solution:

Simply setting 

os.environ["CUDA_VISIBLE_DEVICES"] to the desired GPU does the job
Iffa_Intel
Moderator
163 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
156 Views

Yes, feel free to close it.

Thank you

Iffa_Intel
Moderator
150 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