jupyter notebook pytorch环境安装教程

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

1. 安装 Anaconda

首先，你需要安装 Anaconda，它是一个用于科学计算的包管理器和环境管理器。你可以从以下链接下载适合你的操作系统版本：https://www.anaconda.com/products/individual

2. 创建并激活虚拟环境

打开 Anaconda Prompt，输入以下命令来创建一个名为 pytorch_env 的虚拟环境：

conda create -n pytorch_env

然后，激活该环境：

conda activate pytorch_env

3. 安装 PyTorch

在激活的虚拟环境下，输入以下命令来安装 PyTorch：

conda install pytorch torchvision torchaudio cpuonly -c pytorch

如果你想安装适用于 GPU 的 PyTorch 版本，你需要安装 CUDA 和 cuDNN，并使用以下命令：

conda install pytorch torchvision torchaudio cudatoolkit=<your_cuda_version> -c pytorch

其中，`<your_cuda_version>` 是你安装的 CUDA 版本。

4. 安装 Jupyter Notebook

在激活的虚拟环境下，输入以下命令来安装 Jupyter Notebook：

conda install jupyter

5. 启动 Jupyter Notebook

在激活的虚拟环境下，输入以下命令来启动 Jupyter Notebook：

jupyter notebook

然后，你可以在浏览器中打开 Jupyter Notebook，并在其中创建一个新的 Python 3 笔记本。在笔记本中输入以下代码来测试 PyTorch 是否正确安装：

import torch

x = torch.Tensor([1, 2, 3])
print(x)

如果一切正常，你应该看到输出 `[1, 2, 3]`。