PyTorch не может обнаружить CUDA

Question

Я использую CNN на PyTorch.Функция torch.cuda.is_available () вернула false, и графический процессор не обнаружен.Тем не менее, я могу запустить модель Keras с графическим процессором.Вот моя системная информация:

• ОС: Ubuntu 18.04.3
• Python 3.7.3 (Conda)
• GPU: GTX1080Ti
• Nvidiaдрайвер: 430.50

Когда я проверяю nvidia-smi, вывод говорит, что версия CUDA - 10.1.Однако команда nvcc -V сообщает мне, что это CUDA 9.1.

Я скачал NVIDIA-Linux-x86_64-430.50.run с официального сайта и установил его с помощью командной строки.Я установил CUDA 10.1, используя следующую командную строку, рекомендованную официальным сайтом:

wget http://developer.download.nvidia.com/compute/cuda/10.1/Prod/local_installers/cuda_10.1.243_418.87.00_linux.run
sudo sh cuda_10.1.243_418.87.00_linux.run

Я установил PyTorch через pip install.Что случилось?Заранее спасибо!

Answer 1

Пакет Pytorch 1.2 по умолчанию зависит от CUDA 10.0, но у вас есть CUDA 9.1.Вывод nvidia-smi просто сообщает вам максимальную версию CUDA, которую поддерживает ваш GPU, nvcc дает CUDA, установленную в вашей системе.Кажется, что ваша установка CUDA 10.1 была неудачной.

В дополнение к CUDA 10.0, Pytorch также поддерживает CUDA 9.2, и я обнаружил, что пакет Pytorch, скомпилированный для CUDA 10.0, также работает с CUDA 10.1.Таким образом, вы можете либо обновить установку CUDA до версии 9.2 и установить пакет Pytorch CUDA 9.2 с

pip3 install torch==1.2.0+cu92 torchvision==0.4.0+cu92 -f https://download.pytorch.org/whl/torch_stable.html

, либо получить работающую установку CUDA 10.1.Здесь есть подробные инструкции для Linux . (Обратите внимание, что вам может потребоваться удалить предыдущие установки CUDA перед установкой новой.)