我已经通过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,但未进行任何更改。有人可以帮我吗?
答案 0 :(得分:1)
这似乎是Nvidia容器问题。
将Nvidia容器从18.08降级到18.07后,所有代码都运行良好。
答案 1 :(得分:0)
最好让anaconda自己处理一切。目前,我有pytorch版本0.2,cudatoolkit版本7.5,cudnn版本5.1。工作正常。您可以尝试
conda update --all
它应该解决所有冲突。