Как использовать строку @jit (target = 'cuda') для использования графического процессора - PullRequest
Новая
       9

Как использовать строку @jit (target = 'cuda') для использования графического процессора

0 голосов
/ 27 января 2020

Я пытаюсь ускорить мой код с помощью моего графического процессора, и я нашел строку @jit(target='cuda'), использующую пакет numba, чтобы разрешить использование графического процессора, но мне было интересно, если @jit(target='cuda') выполняет только непосредственную функцию под ним работать с использованием графического процессора или, если это заставит весь код работать на графическом процессоре? Скажите, пожалуйста, каким образом в моем коде должна быть строка: 1. 

@jit(target='cuda')
def func1():
    for x in range(100):
        pass


# then reuse the line


@jit(target='cuda')
def func2():
    for x in range(100):
        pass

или я должен это сделать: 

# only use the line once
@jit(target='cuda')
def func1():
    for x in range(100):
        pass


def func2():
    for x in range(100):
        pass

Если у меня возникли некоторые недопонимания о том, как использовать numba или Jit, пожалуйста, скажите мне, у меня еще есть много, чтобы узнать о компьютерных науках и python.

...