cuda运行时错误（3）：/pytorch/aten/src/THC/THCGeneral.cpp:51处的初始化错误

Question

我开始在终端上运行TensorFlow应用程序。同时，当我在另一个终端上启动Pytorch应用程序时出现错误

THCudaCheck FAIL file=/pytorch/aten/src/THC/THCGeneral.cpp line=51  error=3 : initialization error
terminate called after throwing an instance of 'std::runtime_error'
what():  cuda runtime error (3) : initialization error at /pytorch /aten/src/THC/THCGeneral.cpp:51

Pytorch TensorFlow都在不同的virtualenv上运行

我的环境 -Ubuntu 18.04

-GPU GeForce GTX 1060

-Pytorch env（torch == 1.1.0，torchvision == 0.2.0）

-Tensorflow环境（tensorflow-gpu == 1.15.0）

Pytorch应用程序在启动TensorFlow应用程序之前运行平稳

我停止TensorFlow应用程序并检查

>>> torch.cuda.device_count()
0
>>> torch.cuda.is_available()
False

但错误不会返回

Answer 1

从睡眠模式唤醒Ubuntu后，我遇到了pytorch培训脚本崩溃的同样问题。 Torch无法检测到GPU。似乎Cuda驱动程序在唤醒后无法恢复活动上下文。

重新启动系统解决了该问题。

我在“ conda env ”中运行pytorch。