gpu is not supported in screen sessions
If you activate the tensorflow_p36 environment, and then run a python script that uses tensorflow, you see that the gpu is utilized. However, if you start a screen session, then activate the tensorflow_p36 environment, running the same script will only run on the CPU. Very aggravating when you need to train large scale models that take days (requiring the use of screen/tmux)
There are no comments to display