PJLab-ADG / 3DTrans

An open-source codebase for exploring autonomous driving pre-training
https://bobrown.github.io/Team_3DTrans.github.io/
Apache License 2.0
585 stars 72 forks source link

No module named ‘torch_scatter’ #2

Closed Qzhooooo closed 1 year ago

Qzhooooo commented 1 year ago

When I run the code using the following script, I met a error of ModuleNotFoundError: No module named ‘torch_scatter’ ''' sh ./scripts/UDA/dist_train_uda.sh 4 --cfg_file ./cfgs/DA/waymo_kitti/voxel_rcnn_pre_SN_feat_3.yaml ''' I have installed the code using the INSTALL.md, and please tell me how to fix this error?

sky-fly97 commented 1 year ago

Hello, torch_scatter needs to download .whl for installation. Please note that the version of torch_scatter should be consistent with the version of pytorch and python.