【使用TensorFlowGPU报错】CUDA runtime implicit initialization on GPU:0
我在CPU上能够正常运行,但是当我把自己的程序部署到服务器上面的时候,发现出现错误:
tensorflow.python.framework.errors_impl.InternalError: CUDA runtime implicit initialization on GPU:0 failed. Status: out of memory
参考了这篇文章,报错的原因是我将pytorch和TensorFlow都装在了一个虚拟环境当中了,导致pytorch和TensorFlow冲突。解决办法就是将py
原创
2021-03-29 10:35:18 ·
5316 阅读 ·
0 评论