1. 查看GPU使用情况

    watch -n 1 nvidia-smi[

    ](https://docs.nvidia.com/deeplearning/frameworks/tensorflow-release-notes/index.html)

    1. 输出torch版本

    python
    import torch
    print(torch.__version__)

    1. 输出cuda版本

    print(torch.version.cuda)

    1. 查看cuda和pytorch是否相符

    print(torch.cuda.is_available())

    1. 查看cudnn版本

    print(torch.backends.cudnn.version())

    1. 查看nvidia的信息

    nvidia-smi --query-gpu=name --format=csv,noheader