本机:NVIDIA GeForce RTX 5060笔记本电脑GPU sm_120与当前的PyTorch安装不兼容。实际显卡是NVIDIA GeForce RTX 5060 Laptop GPU,计算能力为sm_120(但实际上是sm_89,因为50系列的计算能力是8.9)
当前的PyTorch安装支持CUDA功能sm_37 sm_50 sm_60 sm_61 sm_70 sm_75 sm_80 sm_86 等。
PyTorch从2.0版本开始支持CUDA 11.8及更高版本,而CUDA 11.8支持sm_89(RTX 40和50系列)
解决方案:三部分
一是新电脑安装cuda
Cuda12.8,匹配显卡5060版本
二是直接从pycharm中conda中直接创建新环境
Python版本:3.12(版本不能太低)
Pytorch下载【官网】,pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu128(对于5060显卡只要找到这个合适的)
【最终版本我装的是torch2.8.0+cu128】
第三步是安装diff-gaussian-rasterization-main和simple_knn的包
终端
# 进入项目根目录
D:
cd D:\python\gaussian-splatting\gaussian-splatting
# 安装simple_knn(带编译过程)
pip install ./submodules/simple-knn
# 安装diff-gaussian-rasterization-main(带编译过程)
pip install ./submodules/diff-gaussian-rasterization-main
第四步不是必需的,根据报错安装需要安装的包
conda uninstall numpy -y
conda clean --all -y
conda install -c conda-forge numpy
有可能需要在代码最前面添加下面这句无视报错
os.environ['KMP_DUPLICATE_LIB_OK'] = 'TRUE'