任何测试 Pytorch 安装的脚本

Question

我已经安装了pytorch，想检查是否有任何脚本来测试安装是否正确，例如是否可以启用CUDA等？

Answer 1

来到你的第一个问题，在你的 python 脚本中......只需添加

import torch

如果出现“ ModuleNotFoundError：没有名为 'torch' 的模块”，那么你的 pytorch 安装尚未完成

你的第二个问题是检查你的 pytorch 是否正在使用 cuda，使用这个

torch.cuda.is_available()

如果你的 pytorch 使用 cuda，这将返回True

Answer 2

如果你从这里安装它，你就做得很好。

检查一下：

import torch
dev = torch.device("cuda") if torch.cuda.is_available() else torch.device("cpu")
print(dev)

如果您的 GPU 安装正确，您应该拥有nvidia-smi. （在 Windows 上它应该在里面C:\Program Files\NVIDIA Corporation\NVSMI）