所有依赖项都安装在 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
最佳答案
我遇到了同样的问题,我能够在构建 SCS-PYTHON 时修复它,如下所示:
python3 setup.py install --scs --gpu --int
如果你查看 setup.py 帮助,它会显示:
GPU code always uses 32 bit ints
python3 setup.py install --help
关于python - 如何安装CVXPY/SCS与GPU一起使用?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/52579531/