Я пытаюсь запустить эту простую строку кода в док-контейнере, который поставляется с Pytorch.
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() возвращает False.
На хост-машине установлены самые последние версии драйверов Nvidia. Pytorch поставляется с Cuda, поэтому не должно быть проблем с несовместимостью.
Что может вызвать эту проблему?
Edit:
@Patel Sunil ответ на этот вопрос отвечает на мой вопрос, но я не сталкивался с этим вопросом в своем поиске, потому что у него широкий вопрос, в то время как мой вопрос касается ошибки времени выполнения / драйвера cuda. Я разместил это как отдельный вопрос для тех, кто сталкивался с этой ошибкой, но не знает, что это за симптом (а именно, забыл использовать nvidia-docker).