本文将为您介绍如何在 PyTorch 教程中使用虚拟环境。

安装 PyTorch

首先，您需要安装 PyTorch。请访问PyTorch 官网获取安装指南。

创建虚拟环境

创建虚拟环境可以避免不同项目之间依赖冲突。以下是在 Python 中创建虚拟环境的步骤：

  1. 打开终端或命令提示符。
  2. 输入 python -m venv myenv 创建一个名为 myenv 的虚拟环境。
  3. 激活虚拟环境：
    • Windows: myenv\Scripts\activate
    • macOS/Linux: source myenv/bin/activate

安装 PyTorch 到虚拟环境

激活虚拟环境后，使用以下命令安装 PyTorch：

pip install torch torchvision torchaudio

使用虚拟环境

现在您可以在虚拟环境中运行 PyTorch 代码了。例如，创建一个名为 hello.py 的文件，并添加以下代码：

import torch

print("Hello, PyTorch!")

运行此代码：

python hello.py

您应该会看到输出 "Hello, PyTorch!"。

附加资源

如果您需要更多帮助，请访问PyTorch 文档。

示例图片

下面是一张 PyTorch 的示例图片：

PyTorch