Получение Torch для распознавания GPU

Question

Как заставить Torch распознавать CUDA на вашей видеокарте?

У меня есть Nvidia GeForce GT 1030, работающая под Ubuntu 18.04, и она заявляет, что поддерживает CUDA , но когда ясначала протестировал Torch, запустив:

virtualenv -p python3.7 .env
. .env/bin/activate
pip install torch
python -c "import torch; print(torch.cuda.is_available())"

он вернул False вместе с предупреждением:

    The NVIDIA driver on your system is too old (found version 9010).
Please update your GPU driver by downloading and installing a new
version from the URL: http://www.nvidia.com/Download/index.aspx
Alternatively, go to: https://pytorch.org to install
a PyTorch version that has been compiled with your version
of the CUDA driver.

Поэтому я запустил все системные обновления и использовал проприетарный установщик драйверов Ubuntu для установкисамый последний драйвер Nvidia-435 для моей карты.

Тем не менее, torch.cuda.is_available() по-прежнему возвращает false, но теперь он не выдает мне никаких предупреждений.

Неправильно ли я настроил Torch или нетмой GPU просто не поддерживает CUDA?

Answer 1

Nevermind. Я говорил слишком рано. Я не перезагружался после переключения драйвера, и, видимо, это сломало nvidia-smi и некоторые другие вещи, которые загружали драйвер CUDA. После перезагрузки Torch теперь распознает поддержку CUDA 10.1.

Answer 2

Да, я проверил эту ссылку , и GT 1030 не совместим.