diff --git a/README.md b/README.md index cd80c6d..dd07b08 100644 --- a/README.md +++ b/README.md @@ -82,6 +82,7 @@ Test availability with: ```python3 import torch +torch.__version__ torch.cuda.is_available() ```