所有依赖项都安装在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
答案 0 :(得分:2)
我遇到了同样的问题,并且能够像下面这样修复SCS-PYTHON:
python3 setup.py install --scs --gpu --int
如果您查看setup.py帮助,它会显示:
GPU代码始终使用32位整数
python3 setup.py install --help