在 Jupyter Notebook 中安装 PyTorch - Shell-Bash

在 Jupyter Notebook 中安装 PyTorch 是进行深度学习任务的关键步骤之一。在本文中，我们将向您展示如何使用 Shell-Bash 在 Jupyter Notebook 中安装 PyTorch。

步骤

以下是在 Jupyter Notebook 中安装 PyTorch 的步骤：

1. 打开 Jupyter Notebook。

2. 在新的 Notebook 中输入以下命令来安装 PyTorch：

!pip install torch torchvision

3. 运行以上命令后，等待几分钟，直到安装完成。在安装期间，您可以认真查看命令行输出，以了解安装过程中是否出现任何错误。

4. 安装完成后，您可以使用以下代码来验证 PyTorch 是否安装成功：

import torch

if torch.cuda.is_available():
    print('CUDA is available.')
else:
    print('CUDA is not available.')

5. 如果上述代码能够正确运行，则说明 PyTorch 已成功安装并可以使用。

结论

在 Jupyter Notebook 中安装 PyTorch 很简单，只需要使用 Shell-Bash 进行安装。如果您遇到安装问题，请尝试搜索相关问题或在 PyTorch 的官方文档中查找帮助章节。