解决docker容器使用ssh连接时，无法调用GPU问题

问题现象

直接使用docker run命令运行tensorflow程序正常
但是用pycharm远程调试时上述容器时运行tensorflow程序失败，提示CUDA错误

问题原因

pycharm远程调试使用的是ssh连接，经测试用ssh远程连接运行也失败
对比docker run和ssh连接两种情况下的环境变量，发现ssh连接少了很多环境变量

解决方案

临时使用环境变量

export $(cat /proc/1/environ |tr '\0' '\n' | xargs -0)

永久使用环境变量

在/etc/profile文件中追加export $(cat /proc/1/environ |tr '\0' '\n' | xargs -0)

参考

https://xuxinkun.github.io/2019/03/12/docker-env/