PyTorch 看不到 GPU (torch.cuda.is_availble() returns False)

Question

我有一个问题

import torch
print(torch.cuda_is_available())

将打印 False，我无法使用可用的 GPU。我已经在 conda 环境中尝试过，我在其中安装了与我拥有的 NVIDIA 驱动程序对应的 PyTorch 版本。我也在 docker 容器中尝试过，我也在那里做过同样的事情。我已经在远程服务器上尝试了这两个选项，但都失败了。我知道我已经安装了正确的驱动程序版本，因为我在安装 PyTorch 之前用 nvcc --version 检查了版本，并且我用 nvidia-smi 检查了 GPU 连接，它在机器上正确显示了 GPU .

此外，我检查了 post 并尝试导出 CUDA_VISIBLE_DEVICES，但没有成功。

在我构建的 docker 容器上，我在服务器上安装了带有 CUDA 10.0 版（用于 conda 环境）和 10.2 版的 NVIDIA V100 GPU。任何帮助或推动正确的方向将不胜感激。谢谢！

Answer 1

对于遇到此问题的其他人，原来我的服务器管理器没有更新服务器的驱动程序。

我切换到另一台服务器，安装了 anaconda，一切开始正常工作，即 torch.cuda.is_available() returns True 在设置了一个新环境后。