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

时间:2018-10-23 10:09:10

标签: pytorch

我已经通过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,但未进行任何更改。有人可以帮我吗?

2 个答案:

答案 0 :(得分:1)

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

答案 1 :(得分:0)

最好让anaconda自己处理一切。目前,我有pytorch版本0.2,cudatoolkit版本7.5,cudnn版本5.1。工作正常。您可以尝试

conda update --all

它应该解决所有冲突。