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

【问题标题】：cuda runtime error (3) : initialization error at /pytorch/aten/src/THC/THCGeneral.cpp:51cuda 运行时错误（3）：/pytorch/aten/src/THC/THCGeneral.cpp:51 处的初始化错误
【发布时间】：2020-09-09 05:54:34
【问题描述】：

我开始在终端上运行 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)

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

我停止 TensorFlow 应用程序并检查

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

但错误不会返回

【问题讨论】：

标签： python-3.x runtime-error pytorch nvidia

【解决方案1】：

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

重启系统解决了这个问题。

我在“conda env”中运行pytorch。

【讨论】：

reboot 也解决了我的问题，但我不认为它可以被视为解决方案。应该有一种方法可以以某种方式软重启 GPU 分配。我仍然找不到这个问题的解决方案。