进入 openmmlab 环境
conda activate openmmlab
设置环境变量为 CUDA 12.4（只影响当前 shell 会话）
export PATH=/usr/local/cuda-12.4/bin:PATHexportLDLIBRARYPATH=/usr/local/cuda−12.4/lib64:PATH export LD_LIBRARY_PATH=/usr/local/cuda-12.4/lib64:PATHexportLDLIBRARYPATH=/usr/local/cuda−12.4/lib64:LD_LIBRARY_PATH
确认 nvcc 正确
which nvcc
nvcc --version