PyTorch 无法看到 GPU（torch.cuda.is_availble() 返回 False）

Question

我有一个问题在哪里

import torch
print(torch.cuda_is_available())

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

另外，我检查了这篇文章并尝试导出CUDA_VISIBLE_DEVICES，但没有运气。

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

Answer 1

对于遇到此问题的其他人，事实证明我的服务器管理器尚未更新服务器的驱动程序。

我切换到不同的服务器，安装了 anaconda，一切开始正常工作，即在设置新环境后torch.cuda.is_available()返回。True