cmd怎么查看自己的pytorch是cpu,还是gpu
时间: 2024-02-28 12:57:40 浏览: 341
pytorch gpu版
在 Windows 系统的命令行中,您可以使用以下命令来检查您的 PyTorch 是否使用了 GPU:
```
python -c "import torch; print(torch.cuda.is_available())"
```
这个命令将会导入 PyTorch 库并检查当前系统中是否有可用的 GPU。如果有可用的 GPU,则 PyTorch 将会使用 GPU,输出为 True,否则 PyTorch 将会使用 CPU,输出为 False。
请注意,您需要在系统中安装了支持 CUDA 的 PyTorch 才能使用 GPU。如果您没有安装支持 CUDA 的 PyTorch,上述命令将会输出 False。
阅读全文