SensatUrban icon indicating copy to clipboard operation
SensatUrban copied to clipboard

about tensorflow version

Open amiltonwong opened this issue 3 years ago • 5 comments

Hi, @QingyongHu ,

According to main_SensatUrban.py and RandLANet.py, tensorflow is used. Which version of tensorflow is needed? 1.1x or 2.x?

Thanks~

amiltonwong avatar Jun 04 '21 12:06 amiltonwong

I am using tensorflow-gpu==1.14 with cuda11.0 and it seems to be okay.

serser avatar Jul 03 '21 18:07 serser

Hi, @amiltonwong @serser. Thanks for your interest in our work.

We use tensorflow-gpu==1.14 with CUDA-9.0, but you are encouraged to try other versions.

Best,

QingyongHu avatar Jul 04 '21 02:07 QingyongHu

TensorFlow 2.x doesn't work for me. 1.14/1.15 works

Remosy avatar Jul 21 '21 16:07 Remosy

I use tensorflow-gpu==1.14 with CUDA-10.0 & cudnn-7.4, it works. And using tensorflow-gpu==1.14 with CUDA-11 can run, but GPU don't work.

SuperKX avatar Sep 02 '21 06:09 SuperKX

tensorflow 2 nightly with tensorflow.compat.v1 works for me. CUDA 11.1 with cudnn 8

xianyu-wang avatar Mar 03 '22 06:03 xianyu-wang