plotly: укажите графический процессор для вывода рисунка - PullRequest
Новая
       32

plotly: укажите графический процессор для вывода рисунка

1 голос
/ 02 августа 2020

У меня два графических процессора на одной машине. Я написал функцию для вывода трехмерной контурной карты с использованием plotly. Один GPU устарел и используется только для мониторной графики. Есть ли способ указать plotly графику только на другом более быстром GPU?

Вот пример кода для визуализации. 

M = np.zeros(64, 64, 64)
X, Y, Z = np.mgrid[0:1:64, 0:1:64, 0:1:64]

try:
    fig = go.Figure(data=go.Volume(
        x=Z.flatten(),
        y=Y.flatten(),
        z=X.flatten(),
        value=M.flatten(),
        opacity=.1,
        surface_count=15,
        cmax=1.,
        cmin=0.
    ))

    # figure IO
    fig.write_image('image.png')
    if show:
        fig.show()

except (OSError, ValueError):
    print('Unable to generate plotly figure')

Edit 1:

CUDA_VISIBLE_DEVICE=0 python myscript.py похоже не помогает. То же касается os.environ('CUDA_VISIBLE_DEVICE') = 0.

...