环境换成你自己的即可
ln -s /home/suyong/anaconda3/envs/openmmlab_zhao/lib/python3.10/site-packages/nvidia/nvjitlink/lib/libnvJitLink.so.12 /home/suyong/anaconda3/envs/openmmlab_zhao/lib/python3.10/site-packages/nvidia/cusparse/lib/libnvJitLink.so.12
export LD_LIBRARY_PATH=/home/suyong/anaconda3/envs/openmmlab_zhao/lib/python3.10/site-packages/nvidia/cusparse/lib:/home/yuan/zhao/cuda/cuda-12.0/lib64:$LD_LIBRARY_PATH