代码来源:https://github.com/city-super/Scaffold-GS
版本设置为autodl中的3090卡,cuda版本11.8,python版本为3.8.20,torch版本为2.0.0+cu118,torchaudio版本为2.0.1+cu118,torchvision版本为0.15.1+cu118,torch-scatter版本为2.1.1+pt20cu118,cudatoolkit版本为11.8。其余包版本和environment.yml中的一致。
torch-scatter直接用指令安装会显示查无此包,直接从网站上按照自己的cuda,torch和python版本下载对应包即可:https://data.pyg.org/whl/
然后修改single_train.sh中的路径,尝试运行。
会出现几个缺少包的报错(我这里是colorama,jaxtyping和cv2),用pip安装即可。
开始运行,首先会下载pth文件:vgg16-397923af.pth,然后输出一堆东西后就会开始训练了。
首次运行截图:
注:一定要注意CUDA,torch和python的版本一致性,优先保证一致,而不是强求和environment.yml内的版本一致。不然可能会出现栈溢出,或者两个simple-knn等两个3DGS自带包无法正确安装的问题。