如何安装 CVXPY / SCS 以与 GPU 一起使用?

Question

所有的依赖项都安装在docker容器中，但是当我使用GPU=True运行示例时，它返回"状态：无界"，并且GPU=False工作正常。scs-python GPU 测试返回"状态：无界"。此外，nvidia-smi显示GPU使用情况。可能是什么问题？

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

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

操作系统：Azure上的Ubuntu 16.04，图像NVIDIA GPU云图像，NVIDIA Tesla P100

Answer 1

我遇到了同样的问题，我能够像这样构建 SCS-PYTHON 来修复它：

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

如果你看一下 setup.py 帮助，它说：

GPU 代码始终使用 32 位整数

python3 setup.py install --help