search

PRESENT-Y / TaG-Net

[IEEE-TMI'23] TaG-Net: Topology-aware Graph Network for Centerline-based Vessel Labeling
GNU General Public License v3.0
34 stars 1 forks source link

请问3090显卡能跑这个代码吗 #2

Open lajiton opened 4 months ago

lajiton commented 4 months ago

您好，拜读了您的文章，觉得效果非常好，想尝试复现，但是我看到您pytorch版本是0.4.1版本的，想请问您是用什么显卡写的，我用3090显卡能不能复现，大概需要改变的是哪些库，再次表示感谢！！！

PRESENT-Y commented 4 months ago

实验使用的显卡是Titan RTX，应该是可以用3090复现的。 具体改变哪些库，我是不太清楚的哇，你可以先尝试一下，遇到什么问题再进行交流。 非常感谢您对我工作的认可，欢迎多多交流~