Нет выхода после использования PyCUDA - PullRequest
Новая
       28

Нет выхода после использования PyCUDA

0 голосов
/ 11 декабря 2018

Я установил PyCUDA, используя pip.Я пробовал это на двух компьютерах.
Один с новой установкой Python 3.7.1 и один с Python 3.6.5.

Все не работает после использования PuCUDA без сообщения об ошибке.

Минимальный пример такой: 

import sys
import pycuda.driver as cuda
import pycuda.autoinit # <-- Comment in order for `print` to work

if __name__ == '__main__':
    print('Print works')
    sys.stdout.write("Sys print works")

Это ничего не печатает, если я не удалю pycuda.autoinit.

Другой пример будетиспользование printf: 

import pycuda.driver as cuda
import pycuda.autoinit
from pycuda.compiler import SourceModule

if __name__ == '__main__':
    mod = SourceModule("""
        #include <stdio.h>

        __global__ void test() {
          printf("I am %d.%d\\n", threadIdx.x, threadIdx.y);
        }
        """)

    func = mod.get_function("test")
    func(block=(4, 4, 1))

Это также не возвращает никакого вывода.

Я думаю, что CUDA завершается неудачно, но ничего не сообщается.

Моя конфигурация: 

+--------------------+--------------------+
|        PC1         |        PC2         |
+--------------------+--------------------+
| Python 3.6.5       | Python 3.7.1       |
| Windows 10         | Windows 10         |
| Cuda toolkit 9     | Cuda toolkit 10    |
| GeForce GTX 1050   | GeForce GTX 1080   |
| Visual Studio 2015 | Visual Studio 2015 |
+--------------------+--------------------+

Драйверы: 

GeForce Game Ready Driver
Version        : 418.91 WHQL
Release Date   : Wed Feb 13, 2019

Я заметил, что это общая проблема , но решения не найдено.

Ответы [ 2 ]

0 голосов
/ 13 февраля 2019

Я не уверен, но надеюсь, что это поможет вам решить проблему.

  1. Проверьте, установлен ли на вашем Windows CUDA инструментарий.
  2. Проверьте, правильно ли установлены PATH окружения для CUDA.
  3. Если вы используете последнюю версию VS, для PyCUDA может быть трудно (или невозможно) работать с ней.Поэтому, если вы хотите установить более старую версию VS дополнительно в вашей текущей системе, после вашей установки вам, возможно, потребуется указать путь к вашему новому (старомодному) компилятору.

Например, вы можете добавить эту строку в функцию compiler в site-packages\pycuda\compiler.py:

options.extend(['-ccbin', 'C:\\Program Files (x86)\\Microsoft Visual Studio 14.0\\VC\\bin\\cl.exe'])

0 голосов
/ 11 декабря 2018

Я не уверен, поможет ли это вам, но подобная проблема была решена переустановкой cuda (с anoconda).

Если вам это нужно, вы можете установить его через conda консоль

conda install -c anaconda cudatoolkit

А чтобы проверить, хорошо ли оно работает,

введите numba -s

Надеюсь, это поможет

...