python-3.x - cuda 运行时错误 (3) : initialization error at/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 和运行在不同的 virtualenvs 上

我的环境 -Ubuntu 18.04

-GPU GeForce GTX 1060

-Pytorch环境(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 从 sleep 模式唤醒后，pytorch 训练脚本崩溃也遇到了同样的问题。 Torch 无法检测到 GPU。看来 Cuda 驱动程序在唤醒后恢复事件上下文有问题。

重启系统解决了问题。

我在“conda env”中运行 pytorch。