OpenMM 7.7.0 与 CUDA 11.6 版本不兼容导致 'CUDA_ERROR_UNSUPPORTED_PTX_VERSION' 错误

如果你在运行 Python 代码时遇到以下错误信息：

Traceback (most recent call last):
  File '/input_kdtest/IR.py', line 201, in <module>
    nvt_sample(parameter, trajectory)
  File '/input_kdtest/IR.py', line 43, in nvt_sample
    simulation = Simulation(pdb.topology, system, integrator, platform=Platform.getPlatformByName('CUDA'), platformProperties={'Precision':'mixed'})
  File '/opt/mamba/lib/python3.10/site-packages/openmm/app/simulation.py', line 103, in __init__
    self.context = mm.Context(self.system, self.integrator, platform, platformProperties)
  File '/opt/mamba/lib/python3.10/site-packages/openmm/openmm.py', line 12378, in __init__
    _openmm.Context_swiginit(self, _openmm.new_Context(*args))
openmm.OpenMMException: Error loading CUDA module: CUDA_ERROR_UNSUPPORTED_PTX_VERSION (222)

这是因为你的 OpenMM 版本 (7.7.0) 与 CUDA 版本 (11.6) 不兼容。

解决方法

要解决这个问题，你需要更新 OpenMM 到支持 CUDA 11.6 的版本。你可以尝试以下方法：

升级 OpenMM 到最新版本：
- 访问 OpenMM 官方网站或代码仓库，查看是否有支持 CUDA 11.6 的最新版本。
- 使用 conda update -c conda-forge openmm 或 pip install --upgrade openmm 命令升级 OpenMM。
查找与 CUDA 11.6 兼容的 OpenMM 版本：
- 查阅 OpenMM 官方文档，找到与你的 CUDA 版本兼容的 OpenMM 版本列表。
- 根据文档说明安装对应版本的 OpenMM。

此外，请确保你的 CUDA 驱动程序和 CUDA 工具包也是最新版本，并且与你安装的 OpenMM 版本兼容。

通过更新 OpenMM 或安装兼容版本，你应该能够解决 'CUDA_ERROR_UNSUPPORTED_PTX_VERSION' 错误并顺利运行代码。

OpenMM 7.7.0 与 CUDA 11.6 版本不兼容导致 'CUDA_ERROR_UNSUPPORTED_PTX_VERSION' 错误