如何查看pytorch是否是gpu版本的
时间: 2023-04-09 07:03:37 浏览: 222
你可以在 Python 中运行以下代码来查看 PyTorch 是否是 GPU 版本:
```python
import torch
print(torch.cuda.is_available())
```
如果输出结果为 True,则表示你安装的是 GPU 版本的 PyTorch。如果输出结果为 False,则表示你安装的是 CPU 版本的 PyTorch。
相关问题
linux查看pytorch是否是gpu版本
可以使用以下命令来查看pytorch是否是gpu版本:
```bash
python -c "import torch; print(torch.cuda.is_available())"
```
如果输出为True,则表示pytorch是gpu版本,否则为cpu版本。
查看anaconda安装的pytorch是否是gpu版本
可以在Python环境中尝试以下代码来检查PyTorch是否安装了GPU版本:
```
import torch
print(torch.cuda.is_available())
```
如果输出结果为True,则表示安装了GPU版本的PyTorch。如果输出结果为False,则表示安装的是CPU版本的PyTorch。