我在CPU上能够正常运行,但是当我把自己的程序部署到服务器上面的时候,发现出现错误:
tensorflow.python.framework.errors_impl.InternalError: CUDA runtime implicit initialization on GPU:0 failed. Status: out of memory
参考了这篇文章,报错的原因是我将pytorch和TensorFlow都装在了一个虚拟环境当中了,导致pytorch和TensorFlow冲突。解决办法就是将pytorch和TensorFlow安装在两个不同的虚拟环境当中。