SensatUrban
SensatUrban copied to clipboard
about tensorflow version
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~
I am using tensorflow-gpu==1.14
with cuda11.0 and it seems to be okay.
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,
TensorFlow 2.x doesn't work for me. 1.14/1.15 works
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.
tensorflow 2 nightly with tensorflow.compat.v1 works for me. CUDA 11.1 with cudnn 8