diff --git a/README.md b/README.md index 6d7185b093..b40fa0d0fe 100644 --- a/README.md +++ b/README.md @@ -22,7 +22,7 @@ If you find our work useful in your research, please consider citing: ### Installation -Install TensorFlow. You may also need to install h5py. The code has been tested with Python 2.7, TensorFlow 1.0.1, CUDA 8.0 and cuDNN 5.1 on Ubuntu 14.04. +Install TensorFlow. You may also need to install h5py. The code has been tested with Python 2.7, TensorFlow 1.0.1, CUDA 8.0 and cuDNN 5.1 on Ubuntu 14.04. If you are using PyTorch, you can find a third-party pytorch implementation here.