Как явно установить номер графического процессора для неявного пакета Python? ⇐ Python

[Anonymous]

Я пытаюсь установить номер графического процессора для неявного пакета в Python. Я попытался установить os.environ['CUDA_VISIBLE_DEVICES']="2", но задача по-прежнему выполняется на графическом процессоре 0.
Вот код, который я пробовал:< /p>

Код: Выделить всё

import os
import implicit

os.environ['CUDA_VISIBLE_DEVICES'] = "2"

model = implicit.als.AlternatingLeastSquares(factors=50)

model.fit(train_user_items)

Несмотря на установку переменной среды, задача по-прежнему выполняется на графическом процессоре 0. Как я могу гарантировать, что неявный пакет использует указанный графический процессор?
Дополнительно Информация:

• неявная версия: 0.7.2 (собрана из исходного кода)
• Python: 3.11.2
• CUDA: 12.3
• ОС: Debian GNU/Linux 12

Будем очень благодарны за любую помощь или предложения.

Подробнее здесь: https://stackoverflow.com/questions/793 ... it-package