Python torch.cuda.is_available（）连续返回False

我现在在学校的一个实验室里，我们使用GPU服务器来完成任务。我试过运行

torch.cuda.is_available（）

函数，但尽管我试过这个和那个，它还是不断返回False

我所说的“这个和那个”，特别是指我已经尝试过安装和重新安装PyTorch，我通过运行

nvidia smi

命令确保CUDA确实可用，我还检查了

CUDA\u VISIBLE\u DEVICES

环境变量，并确保它被正确分配

GPU设备为

0

或

1

，我的环境变量当前设置为

1

目前我拥有的规格是：

操作系统：Ubuntu 16.04

中大：9.0176

Python:3.7.0

cuDNN:9.0-v7.0.5

Nvidia驱动程序：384.81

有没有人经历过这个问题？如果是，它是如何解决的

编辑

我目前已经按照建议从Github源代码重新安装了PyTorch，它解决了我的问题。然而，如果有人能提供一些关于为什么会出现这些问题的见解，对我和整个社区都会有很大帮助。多谢各位

编辑2

正如@khashit在评论中所建议的那样，我运行了位于

pytorch/torch/utils/

中的

collect_env.py

，得到了以下输出：

Collecting Environment information...
PyTorch version: 1.3.0a0+6a4ca9a
is debug build: No
CUDA used to build PyTorch: 9.0.176

OS: Ubuntu 16.04.3 LTS
GCC version: (Ubuntu 5.4.0-6ubuntu1~16.04.10) 5.4.0 20160609
CMake version: version 3.14.0

Python version: 3.7
Is CUDA available: No
CUDA runtime version: 9.0.176
GPU models and configuration:
GPU 0: TITAN Xp
GPU 1: TITAN Xp

Nvidia driver version: 384.81
cuDNN version: Could not collect

Versions of relevant libraries:
[pip3] numpy==1.17.2
[pip3] numpydoc=0.8.0
[pip3] torch=1.3.0a0+6a4ca9a
[pip3] torch-cluster=1.4.4
[pip3] torch-geometric=1.3.1
[pip3] torch-scatter==1.3.1
[pip3] torch-sparse==0.4.0
[pip3] torch-spline-conv==1.1.0
[pip3] torchvision==0.4.0
[conda] blas                    1.0                    mkl
[conda] mkl                     2019.4                 243
[conda] mkl-include             2019.4                 243
[conda] mkl-service             2.3.0       py37he904b0f_0
[conda] mkl_fft                 1.0.14      py37ha843d7b_0
[conda] mkl_random              1.1.0       py37hd6b4f25_0
[conda] torch                   1.2.0               pypi_0     pypi
[conda] torch-cluster           1.4.4               pypi_0     pypi
[conda] torch-geometric         1.3.1               pypi_0     pypi
[conda] torch-scatter           1.3.1               pypi_0     pypi
[conda] torch-sparse            0.4.0               pypi_0     pypi
[conda] torch-spline-conv       1.1.0               pypi_0     pypi
[conda] torchvision             0.4.0               pypi_0     pypi

---

我对此不太熟悉，也不确定是否有任何信息表明存在潜在问题。如果有，有人能指出吗？谢谢。

检查的输出，它将为您提供更多信息。检查您安装了多少nvidia驱动程序，我遇到了相同的问题，原因是我安装了两个nvidia驱动程序，因此当我删除一个后重新安装Pytorch时，它工作正常