时间戳

首页 > 解决方案 > 如何安装 CVXPY / SCS 以与 GPU 一起使用？

python - 如何安装 CVXPY / SCS 以与 GPU 一起使用？

问题描述

所有依赖项都安装在 docker 容器中，但是当我使用 GPU=True 运行示例时，它返回“Status: Unbounded”，并且使用 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 Cloud 映像，NVIDIA Tesla P100

标签： pythonazuredockernvidiacvxpy

解决方案

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

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

如果您查看 setup.py 帮助，它会说：

GPU 代码始终使用 32 位整数

python3 setup.py install --help

推荐阅读