如何查看pytorch是否是gpu版本的
时间: 2023-04-09 16:03:37 浏览: 301
pytorch安装教程gpu.rar
你可以在 Python 中运行以下代码来查看 PyTorch 是否是 GPU 版本:
```python
import torch
print(torch.cuda.is_available())
```
如果输出结果为 True,则表示你安装的是 GPU 版本的 PyTorch。如果输出结果为 False,则表示你安装的是 CPU 版本的 PyTorch。
阅读全文