docker 容器中的 Cuda Runtime/Driver 不兼容
Cuda Runtime/Driver incompatibility in docker container
我正在尝试 运行 在 Pytorch 附带的 docker 容器中添加这一行简单的代码。
import torch
torch.cuda.set_device(0)
我收到这个错误：
RuntimeError: cuda runtime error (35) : CUDA driver version is insufficient for CUDA runtime version at torch/csrc/cuda/Module.cpp:32
运行 torch.cuda.is_available() returns False.
主机具有最新的 Nvidia 驱动程序。 Pytorch 附带 Cuda，因此应该没有不兼容问题。
什么可能导致此问题？
编辑：
@Patel Sunil 对 this question 的回答回答了我的问题，但我在搜索中没有遇到这个问题，因为他们的问题很广泛，而我的问题是针对 cuda runtime/driver 错误的。对于那些遇到此错误但不知道它是什么症状的人（即忘记使用 nvidia-docker），我将其作为一个单独的问题发布。
问题是我 运行 是 docker 的容器，而不是 nvidia-docker。 运行 带有 nvidia-docker 的 docker 容器修复了问题。
我正在尝试 运行 在 Pytorch 附带的 docker 容器中添加这一行简单的代码。
import torch
torch.cuda.set_device(0)
我收到这个错误：
RuntimeError: cuda runtime error (35) : CUDA driver version is insufficient for CUDA runtime version at torch/csrc/cuda/Module.cpp:32
运行 torch.cuda.is_available() returns False.
主机具有最新的 Nvidia 驱动程序。 Pytorch 附带 Cuda，因此应该没有不兼容问题。
什么可能导致此问题？
编辑： @Patel Sunil 对 this question 的回答回答了我的问题，但我在搜索中没有遇到这个问题，因为他们的问题很广泛，而我的问题是针对 cuda runtime/driver 错误的。对于那些遇到此错误但不知道它是什么症状的人（即忘记使用 nvidia-docker），我将其作为一个单独的问题发布。
问题是我 运行 是 docker 的容器，而不是 nvidia-docker。 运行 带有 nvidia-docker 的 docker 容器修复了问题。