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

How to install CVXPY / SCS to use with GPU?

所有依赖项都安装在 docker 容器中,但是当我 运行 GPU=True 的示例 returns "Status: Unbounded" 和 GPU=False它工作正常。 scs-python GPU 测试 return "Status: Unbounded"。此外,nvidia-smi 显示 GPU 使用情况。可能是什么问题?

Docker 文件:https://gist.github.com/Cerebrock/eef5d33ece1782d3efdf0b35d29fb48e

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

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

我遇到了同样的问题,我能够在构建 SCS-PYTHON 时修复它,例如:

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

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

GPU code always uses 32 bit ints

python3 setup.py install --help