3DCDC-Phys Train train.py Test predict.py Environment Tensorflow 2.0+ conda create -n tf-gpu tensorflow-gpu cudatoolkit=10.1 -- this command takes care of both CUDA and TF environments. conda activate xxx