如何查看PyTorch是否在GPU上运行CUDA

流程图

pie
    title 流程图
    "Step 1" : 安装PyTorch
    "Step 2" : 导入PyTorch库
    "Step 3" : 检查GPU是否可用
    "Step 4" : 查看当前PyTorch是否在GPU上运行

步骤和代码示例

Step 1: 安装PyTorch

在命令行中输入以下代码：

pip install torch torchvision

Step 2: 导入PyTorch库

在Python脚本中导入PyTorch库：

import torch

Step 3: 检查GPU是否可用

使用以下代码检查是否有可用的GPU：

if torch.cuda.is_available():
    print('GPU is available!')
else:
    print('No GPU available, using CPU instead.')

Step 4: 查看当前PyTorch是否在GPU上运行

运行以下代码查看PyTorch当前是否在GPU上运行：

device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
print('PyTorch is running on:', device)

类图示例

classDiagram
    class PyTorch{
        - is_available()
    }
    PyTorch : + device()

通过以上步骤，你可以轻松地检查PyTorch是否在GPU上运行CUDA。希望这篇文章对你有所帮助！如果有任何疑问，请随时向我提问。祝你编程愉快！