使用Cellbender时无法检测CUDA 12的技术问题求助

问题核心 #

Cellbender依赖PyTorch的torch.cuda.is_available()判断CUDA可用性，报错说明当前环境的PyTorch未正确关联CUDA驱动。

排查与解决步骤 #

• 第一步：验证PyTorch的CUDA支持
  在激活Cellbender的conda环境中执行：

  import torch
  print(torch.cuda.is_available())
  print(torch.version.cuda)
  

  如果输出False，说明PyTorch未安装CUDA版本。

• 第二步：重装适配CUDA 12.x的PyTorch
  由于CUDA 12.2属于较新版本，选择PyTorch官方支持的兼容版本（如12.1）进行安装：

  • Conda安装命令：

    conda install pytorch torchvision torchaudio pytorch-cuda=12.1 -c pytorch -c nvidia
    

  • Pip安装命令：

    pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121
    

  注意：必须在Cellbender所在的conda环境中执行安装。

• 第三步：再次验证CUDA可用性
  重新运行第一步的验证代码，确认输出True及对应的CUDA版本（如12.1）。

• 第四步：重新执行Cellbender命令
  在激活的环境中运行原命令：

  cellbender remove-background --input raw_feature_bc_matrix_pDRGN_d0_r1_50.h5 --output output.h5 --cuda --expected-cells 20000 --total-droplets-included 25000 --fpr 0.01 --epochs 150
  

额外排查点 #

• 确保始终激活正确的conda环境，避免混用全局Python环境
• 确认nvidia-smi显示的驱动版本（535.54.03）支持PyTorch安装的CUDA版本（12.1兼容12.2）
• 若Cellbender为源码安装，需在PyTorch环境重新执行pip install -e .

内容的提问来源于stack exchange，提问作者nikhil int