【python报错】3dgs在NVIDIA GeForce RTX 5060运行

本机: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'

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值