possible to add --nv option when running tensorflow GPU container directly? #687
Replies: 3 comments 13 replies
-
You'd want to add it to singularity options in the container.yaml you install from, if I remember correctly. |
Beta Was this translation helpful? Give feedback.
-
Here are differences with the First I also set differences as root:
Differences as non root
Could |
Beta Was this translation helpful? Give feedback.
-
So on RHEL 8 with Anaconda Python 3.11.8 and shpc 0.1.30, running: then the module file shows as just Running
Line 97: |
Beta Was this translation helpful? Give feedback.
-
One of the cool things I see is that one can run commands from inside a container without having to run the
singularity
(orapptainer
) command. For Tensorflow with GPU support, to get to the GPUs I have to run the full cmd:Ideally, I'd like to run any of the tensorflow-* cmds directly:
However the GPU's are not seen by TF this way. Running
python
thenresults in:
Beta Was this translation helpful? Give feedback.
All reactions