最新消息:雨落星辰是一个专注网站SEO优化、网站SEO诊断、搜索引擎研究、网络营销推广、网站策划运营及站长类的自媒体原创博客

pytorch - Compatibility Issue with torch_scatter - Stack Overflow

programmeradmin2浏览0评论

I use Torch 2.4.1, Torchvision 0.19.1, and CUDA 11.8 for my model. Torch-scatter 2.12, which I have already installed, is incompatible with these. How can I determine which version is suitable?

I want to use torch-scatter in this part of my code:

# collect information
point_dict = Dict(feat=torch_scatter.segment_csr(self.proj(point.feat)[indices], idx_ptr, reduce=self.reduce),
发布评论

评论列表(0)

  1. 暂无评论