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

标签 python azure docker nvidia cvxpy

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

相关文章:

python - 这里 "if False:"的作用是什么?

python - Conda安装包ModuleNotFoundError : No module named 'conda'

python - 使用 numpy 在 Pygame 中进行更高效的风洞模拟

azure - 使用私有(private)存储帐户创建 SQL Server 漏洞评估资源失败

php - 在Docker容器中运行laravel环境

node.js - 如何从 Nodejs 应用程序容器使用 mongodump 命令?

python - 在 Django 模型中存储和转义 Django 标签和过滤器

azure - ##[错误]操作返回无效状态代码 'Forbidden' 错误 Azure 管道运行 Set-AzSqlServerActiveDirectoryAdministrator

Azure逻辑应用程序不为Google工作表中的每个实例插入一行

php - Docker 容器与网关的连接被拒绝