diff --git a/README.rst b/README.rst
index 6e305ad..ee2096d 100644
--- a/README.rst
+++ b/README.rst
@@ -103,6 +103,8 @@ reference. Functionality provided by the library includes:
    # Train a model
    train.train_model(train_args, m, (train_loader, val_loader), store=out_store)
 
+**Note**: ``robustness`` requires PyTorch to be installed with CUDA support.
+
 Pretrained models
 --------