Giters

songtaohe / Sat2Graph

Sat2Graph: Road Graph Extraction through Graph-Tensor Encoding

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

training environment

EmilioTyl opened this issue · comments

commented

Hi, really cool work here. Could you specify the training environment for this project? i understand that you are usign tensorflow 14.0 but is it using any specific cuda and cudnn ?

commented

Hi, we updated the environment of our server long time ago. If I remember correctly, we were using tensorflow 1.13.2 with cuda 10.0 and cudnn 7.4.2. The new models were trained with tensorflow 1.15 with cuda 10.0 and cudnn 7.6.4. We don't have anything in our code that relies on some specific cuda/cudnn version. So the code should work with both of these two environments and other cuda/cudnn versions.