diff --git a/README.md b/README.md index 558d7b9..01dbb54 100644 --- a/README.md +++ b/README.md @@ -19,10 +19,10 @@ Many applications in robotics and human-computer interaction can benefit from un ### Installation -Install TensorFlow. The code is tested under TF1.9 GPU version and Python 3.5 on Ubuntu 16.04. There are also some dependencies for a few Python libraries for data processing and visualizations like `cv2`. It's highly recommended that you have access to GPUs. +Install TensorFlow. The code is tested under TF1.9.0 GPU version, g++ 5.4.0, CUDA 9.0 and Python 3.5 on Ubuntu 16.04. There are also some dependencies for a few Python libraries for data processing and visualizations like `cv2`. It's highly recommended that you have access to GPUs. #### Compile Customized TF Operators -The TF operators are included under `tf_ops`, you need to compile them first by `make` under each ops subfolder (check `Makefile`). Update `arch` in the Makefiles for different CUDA Compute Capability that suits your GPU if necessary. The code is tested under TF1.9.0. +The TF operators are included under `tf_ops`, you need to compile them first by `make` under each ops subfolder (check `Makefile`). Update `arch` in the Makefiles for different CUDA Compute Capability that suits your GPU if necessary. ### Usage