跳到主要内容

验证 PyTorch 安装

安装完成后,打开 Python 终端,输入:

import torch
print(torch.__version__)
print(torch.cuda.is_available()) # Windows CUDA 用户应为 True

如输出 PyTorch 版本号且 torch.cuda.is_available() 为 True,说明 GPU 版本安装成功。