PyTorch и CUDA драйвер

Question

У меня установлена ​​CUDA 9.2.Например:

(base) c:\>nvcc --version
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2018 NVIDIA Corporation
Built on Wed_Apr_11_23:16:30_Central_Daylight_Time_2018
Cuda compilation tools, release 9.2, V9.2.88

Я установил PyTorch на Windows 10, используя:

conda install pytorch cuda92 -c pytorch
pip3 install torchvision

Я запустил тестовый скрипт:

(base) c:\>python
Python 3.6.5 |Anaconda custom (64-bit)| (default, Mar 29 2018, 13:32:41) [MSC v.1900 64 bit (AMD64)] on win32
Type "help", "copyright", "credits" or "license" for more information.
>>> from __future__ import print_function
>>> import torch
>>> x = torch.rand(5, 3)
>>> print(x)
tensor([[0.7041, 0.5685, 0.4036],
        [0.3089, 0.5286, 0.3245],
        [0.3504, 0.8638, 0.1118],
        [0.6517, 0.9209, 0.6801],
        [0.0315, 0.1923, 0.8720]])
>>> quit()

Так что, так хорошо.Затем я запустил:

(base) c:\>python
Python 3.6.5 |Anaconda custom (64-bit)| (default, Mar 29 2018, 13:32:41) [MSC v.1900 64 bit (AMD64)] on win32
Type "help", "copyright", "credits" or "license" for more information.
>>> import torch
>>> torch.cuda.is_available()
False
>>>

Почему PyTorch сказал, что CUDA недоступна?

Charles

Answer 1

Официальный вызов PyTorch для использования CUDA 9.0, и я бы предложил то же самое.В других случаях иногда возникают проблемы со сборкой, которые приводят к тому, что «CUDA не обнаружен». Поэтому при использовании PyTorch лучше всего использовать CUDA 9.0 и CuDnn 7. Я добавлю ссылку, где вы можете легко установить Cuda 9.0 и CuDnn 7.

https://yangcha.github.io/CUDA90/

Answer 2

У меня была похожая проблема, вы должны проверить на панели управления NVIDIA, что ваша карта выбрана по умолчанию.

Answer 3

С тех пор как https://github.com/pytorch/pytorch/releases/tag/v0.3.1, Бинарные выпуски PyTorch убрали поддержку старых графических процессоров с поддержкой CUDA 3.0.Согласно https://en.wikipedia.org/wiki/CUDA, вычислительные возможности Quadro K3000M равны 3,0.

Следовательно, вам, возможно, придется собирать Pytorch из исходного кода или пробовать другие пакеты.Пожалуйста, обратитесь к этой теме для получения дополнительной информации - https://discuss.pytorch.org/t/pytorch-no-longer-supports-this-gpu-because-it-is-too-old/13803.