Pytorch (0.4.1) 找不到 GPU (Nvidia V100)

Question

我已经通过 anaconda 安装了带有 CUDA 9.0 的 Pytorch(0.4.1)。
我还安装了 Nvidia 驱动程序 (390.30)。
通过nvidia-smi，我可以看到所有GPU（V100）的情况。
但是在调用 torch.cuda.is_avaliable() 时我总是得到 False。
我也尝试将CUDA版本从9.0降级到8.0，但情况还是一样。
尝试再次安装 Pytorch，但没有任何改变。谁能帮帮我好吗？

Answer 1

这似乎是 Nvidia 容器问题。
将 Nvidia 容器从 18.08 降级到 18.07 后，所有代码运行良好。

Answer 2

最好让 anaconda 自己处理所有事情。暂时我有pytorch 0.2版，cudatoolkit 7.5版，cudnn 5.1版。它工作正常。你可以试试

conda update --all

它应该解决所有的冲突。