如何在jupyter notebook中检测pytorch是否成功

安装了 PyTorch 后，可以在 Jupyter Notebook 中运行以下代码来检测 PyTorch 是否成功安装：

import torch

print(torch.__version__)

if torch.cuda.is_available():
    print('CUDA is available')
else:
    print('CUDA is not available')

如果输出了 PyTorch 的版本号和 CUDA 是否可用的信息，则说明 PyTorch 安装成功。