如何告诉PyCUDA使用哪个GPU?

杰森·道尔顿

我的机器上有两张NVidia卡,并且都支持CUDA。当我运行示例脚本以开始使用PyCUDA时,请参见此处:http ://documen.tician.de/pycuda/我得到了错误

nvcc fatal   : Value 'sm_30' is not defined for option 'gpu-architecture'

我的计算GPU是计算能力3.0,因此sm_30应该是nvcc编译器的正确选项。我的图形GPU仅为CC 1.2,所以我想可能就是问题所在。我已经安装了适用于Linux的CUDA 5.0版本,没有任何错误,以及所有编译器组件和python组件。

有没有办法明确告诉PyCUDA使用哪个GPU?

罗伯特·克罗维拉

nvcc不会根据您已安装的特定GPU进行投诉。它将针对您要求编译的任何GPU类型进行编译。问题是您在指定sm_30选项--gpu-architecture指定了哪个无效选项--gpu-code

你应该compute_30为之--gpu-architecturesm_30--gpu-code

另外,请确保您使用的是正确的nvcc,并且不会无意中使用某些旧版本的CUDA工具包。

解决了编译问题后,CUDA_DEVICEpycuda将观察到一个环境变量,以选择特定的已安装GPU。

这里

CUDA_DEVICE=2 python my-script.py

顺便说一下,其他人遇到了您的问题。您确定您没有使用PyCUDA使用的CUDA工具包的旧版本吗?

本文收集自互联网,转载请注明来源。

如有侵权,请联系 [email protected] 删除。

编辑于
0

我来说两句

0 条评论
登录 后参与评论

相关文章

如何告诉Restkit使用哪个实体映射

如何告诉 Flask (.flaskenv) 使用哪个 python

如何告诉软件使用哪个输出?

我如何告诉Windows使用什么GPU

如何告诉PyTorch不要使用GPU?

使用指针从GPU复制pyCuda数组

如何告诉 Flux.jl 不使用 GPU?

运行tensorflow时如何指定使用哪个GPU?

如何确定ubuntu16.04 LTS中使用的是哪个GPU?

如何告诉readthedocs加载哪个需求文件?

如何告诉xargs选择哪个参数?

使用AutoHotKey从WSL启动GUI程序:如何告诉cmd使用哪个$ DISPLAY变量?

如何从gpu内存地址创建PyCUDA GPUArray?

如何告诉Timber哪个日志调用是针对哪个树的?

哪个进程正在使用GPU?

angular-cli:如何告诉browsersync(sockjs-node)使用哪个端口

如何告诉Spring Boot可执行jar使用哪个主类?

如何告诉react-native在多个中使用哪个JDK

如何告诉作曲家给定作曲家存储库使用哪个密钥?

如何告诉Swift在默认的可选参数之间使用哪个函数签名

我如何告诉我的Dockerfile在docker-compose的RUN命令中使用哪个DNS?

在 R 中使用 t.test 更大的人群是哪个?如何告诉函数?

如何使用PyCUDA处理python列表?

如何在多处理中使用 PyCuda?

如何使用 PyCuda mem_alloc_pitch()

当您使用jspm安装githup存储库时,如何告诉它要获取哪个zip版本,或克隆存储库

如何在不配置新工具链的情况下告诉Bazel使用哪个编译器

如何告诉Eureka Client要注册哪个端口(Spring Cloud)

如何告诉Dagger 2基于X实例化哪个实现?