Update README.rst

MadryLab · Oct 30, 2019 · e1c389d · e1c389d
1 parent 83fd367
commit e1c389d
Showing 1 changed file with 2 additions and 0 deletions.
diff --git 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
 --------