Pytorch на новой установке человечности не может найти мой GPU
Мне установили последнюю человечность 18 только час назад. У меня есть новейшая анаконда 3,7 установленных, возможно, 20 минут назад. Затем я продолжил использовать
conda install pytorch torchvision -c pytorch
но когда я открываю использование python3, оно все еще дает мне
alex@alex:~$ python3
Python 3.7.0 (default, Jun 28 2018, 13:15:42)
[GCC 7.2.0] :: Anaconda, Inc. on linux
Type "help", "copyright", "credits" or "license" for more information.
>>> import torch
>>> torch.cuda.is_available()
False
>>>
Безусловно, у меня действительно есть gpu в моем компьютере:
alex@alex:~$ lspci | grep VGA
01:00.0 VGA compatible controller: NVIDIA Corporation GP107 [GeForce GTX 1050 Ti] (rev a1)
Я также перезапустил свой компьютер после установки pytorch, и это ничего не изменило. Это - моя вторая попытка того, чтобы заставлять GPU работать над pytorch, моя последняя попытка была на Lubuntu несколько дней назад с помощью установки зернышка, и я установил некоторые драйверы Nvidia, но все еще не мог заставить это работать так, я запустил снова.
2 ответа
Просто найденный решением, возможно.... Проверить https://ubuntuforums.org/showthread.php? t=2377560 для деталей.
Возможно, у Вас есть гибридная видеокарта, и это может вызвать неправильную установку драйвера. Также проверьте, что безопасная начальная загрузка отключена в BIOS.
Выполненный в терминале:
sudo apt purge nvidia*
sudo apt update
sudo ubuntu-drivers autoinstall
sudo apt full-upgrade
Перезагрузка и проверка снова.
Поскольку я вижу, что Вы делаете "нормальную" установку PyTorch, который является в данный момент CUDA 9.
Я просто проверил документацию NVIDIA CUDA 9, Ubuntu 18 не поддерживается в этом случае. Если Вы хотите использовать CUDA на Ubuntu 18, то необходимо использовать CUDA 10 согласно документации.