進入 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:PATHexportLDL?IBRARYP?ATH=/usr/local/cuda?12.4/lib64:LD_LIBRARY_PATH
確認 nvcc 正確
which nvcc
nvcc --version