存在两个cuda环境,在conda中切换到另一个
进入 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