WSL2中HOOMD-blue GPU运行报错‘invalid device ordinal’求助

错误信息 #

RuntimeError: CUDA Error: invalid device ordinal before /hoomd/Autotuner.h:497 

环境配置 #

• WSL2 + Ubuntu 22.04
• NVIDIA GPU：RTX 4060（已安装最新驱动）
• WSL内nvidia-smi可正常检测GPU
• 通过micromamba install hoomd=5.2.0安装HOOMD
• CUDA Toolkit版本：12.2（也曾尝试11.8）

测试脚本 #

import hoomd
device = hoomd.device.GPU()  # 也曾显式指定device_id=0

调用GPU.get_available_devices()返回：

['[0]NVIDIA GeForce RTX 2060  30 SM_7.5 @ 1.68 GHz, 6143 MiB DRAM']

可见GPU已被检测到，但初始化设备时立即触发‘invalid device ordinal’错误。

已尝试的解决方法 #

• 设置环境变量CUDA_VISIBLE_DEVICES=0
• 显式传递device_id=0参数
• 升级/降级HOOMD及CUDA版本
• 使用conda和micromamba两种方式安装HOOMD
• 仅运行hoomd.device.GPU()的最小示例

以上方法均无效，仅切换为hoomd.device.CPU()可正常运行，但需要GPU加速。

问题 #

1. HOOMD-blue在WSL2环境下是否完全支持GPU？
2. 若支持，如何修复该‘invalid device ordinal’错误（该错误似乎在使用邻居列表时出现）？
3. 若不支持，是否有已知的替代方案（除切换到CPU或原生Windows/Linux外）？

解答 #

1. HOOMD-blue官方明确支持WSL2环境的GPU加速，但要求CUDA版本、NVIDIA驱动版本与HOOMD的编译版本严格匹配，预编译包可能存在特定环境的适配差异。

2. 针对该错误，可尝试以下修复方案：

   • 手动初始化CUDA上下文：在导入HOOMD前先通过CUDA工具库绑定设备，确保上下文一致性：

     import cupy
     cupy.cuda.Device(0).use()
     import hoomd
     device = hoomd.device.GPU(device_id=0)
     

   • 禁用自动调谐器：错误提示指向Autotuner模块，尝试关闭自动调谐功能：

     device = hoomd.device.GPU(device_id=0, autotune=False)
     

   • 从源码编译HOOMD：放弃预编译包，根据WSL2内的CUDA版本源码编译HOOMD，确保编译环境与运行环境完全匹配。
   • 验证WSL2 GPU直通状态：在Windows终端运行wsl --status确认WSL版本为2且GPU虚拟化已启用；同时检查Windows端NVIDIA驱动是否为WSL专用版本（通用驱动可能存在兼容性问题）。

3. 若上述方案无效，可尝试：

   • 使用Docker容器运行HOOMD：选择官方提供的带GPU支持的HOOMD镜像，利用Docker的GPU直通机制规避环境配置冲突。
   • 强制启用GPU直通：在Windows终端运行wsl --distribution Ubuntu-22.04 --gpu all启动WSL，确保所有GPU设备被正确映射。

内容的提问来源于stack exchange，提问作者Eric Ortiz Vazquez