Debugging using docker interpreter gpu supported

Hi all,

I'm trying to debug using docker interpreter but I keep facing GPU support problems. When running nvidia-docker (from terminal) tensorflow recognizes the GPU devices as expected but it is not the case when I'm using the same image as a docker interpreter.

From what I read at the net passing --runtime=nvidia might solve this issue. I didn't see where it is possible to pass arguments calling docker as part of the configuration file. 

Is there any way to solve this issue? Is there any plan to integrate nvidia-docker as part of remote debugging option?

 

Cheers,

Aviv

0
5 comments

Hey, you can pass the --runtime option according to this comment https://youtrack.jetbrains.com/issue/PY-26429#focus=Comments-27-4026288.0-0

Give it a try, let us know if you have any issues.

0

Hi Andrey Resler

I tried what is written in the link you provided but it didn't work for me. I can't see the run options as presented in the example, tried to pass the args as env variables.

here is an example of the configuration file:

 

I just want to mention, if I use nvidia-docker from terminal the exact image works, and all GPUs been recognized.

0

I will try to update my pycharm version and will have it a go once again.

0

Andrey Resler now it works! thank you so much!

0

Please sign in to leave a comment.