Как установить CVXPY / SCS для использования с графическим процессором?

Все зависимости установлены в контейнере докеров, но когда я запускаю пример с GPU = True, он возвращает «Status: Unbounded», а с GPU = False все работает нормально. Тесты scs-python GPU возвращают «Статус: неограничен». Кроме того, nvidia-smi отображает использование графического процессора. В чем может быть проблема?

Dockerfile: https://gist.github.com/Cerebrock/eef5d33ece1782d3efdf0b35d29fb48e

Пример: https://gist.github.com/Cerebrock/885023ec2faf1d1c6df12df9bf9b

ОС: Ubuntu 16.04 в Azure, образ NVIDIA GPU Cloud, NVIDIA Tesla P100


azure python docker nvidia cvxpy
person Cerebrock    schedule 30.09.2018    source источник

Ответы (1)


arrow_upward
2
arrow_downward

У меня была такая же проблема, и я смог исправить ее, построив SCS-PYTHON, например:

python3 setup.py install --scs --gpu --int

Если вы посмотрите на справку setup.py, там написано:

Код графического процессора всегда использует 32-битные целые числа

python3 setup.py install --help
person ruloweb    schedule 01.10.2018
comment
Есть ли способ сделать это через pip? - person Hennich; 06.12.2018
comment
Вы можете установить его с помощью pip, но графический процессор не будет включен. - person ruloweb; 08.12.2018