如何安装CVXPY / SCS与GPU配合使用?

时间:2018-09-30 15:51:58

标签: python azure docker nvidia cvxpy

所有依赖项都安装在docker容器中,但是当我使用GPU = True运行示例时,它返回“状态:无界”,而使用GPU = False,则可以正常工作。 scs-python GPU测试返回“状态:无界”。此外,nvidia-smi还会显示GPU使用情况。可能是什么问题?

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

示例:https://gist.github.com/Cerebrock/885023ec2faf1d1c6bf9b1a4df12df36

OS:Azure上的Ubuntu 16.04,映像NVIDIA GPU Cloud映像,NVIDIA Tesla P100

1 个答案:

答案 0 :(得分:2)

我遇到了同样的问题,并且能够像下面这样修复SCS-PYTHON:

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

如果您查看setup.py帮助,它会显示:

  

GPU代码始终使用32位整数

python3 setup.py install --help