如何安装 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
所有依赖项都安装在 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