在虚拟环境中安装 PyTorch - Shell/Bash

如果你是一个 Python 开发者，你进入了一个新的项目，你需要使用 PyTorch ，但是你不知道如何在虚拟环境中安装它，那么这篇文章就是为你准备的。

创建虚拟环境

首先，你需要创建一个新的 Python 环境，这是一个可以隔离安装的环境，使得你的各个项目可以使用不同版本的 Python 包，同时也可以防止与主 Python 环境的冲突。

# Mac/Linux
python3 -m venv myenv
# Windows
python -m venv myenv

在上述命令中，myenv 是你的虚拟环境的名称。你可以在任何地方创建虚拟环境，但最好是在你的项目目录中创建。

激活虚拟环境

安装 PyTorch 前，你需要在命令提示符中激活虚拟环境。

# Mac/Linux
source myenv/bin/activate
# Windows
myenv\Scripts\activate.bat

此时你应该看到你的终端提示符发生了变化，并显示了你的虚拟环境名称。

安装 PyTorch

现在你的虚拟环境已经准备好了，你可以安装 PyTorch。

如果你使用的是 CPU-only 版本的 PyTorch，可以运行以下命令：

pip install torch

如果你使用的是支持 CUDA 的 PyTorch （需要 NVIDIA 显卡），则可以运行以下命令：

# CPU version
pip install torch
# CUDA version
pip install torch torchvision torchaudio -f https://download.pytorch.org/whl/cu111/torch_stable.html

测试

安装完成后，你可以使用以下代码片段测试 PyTorch 的安装情况：

import torch

print(torch.__version__)
print(torch.cuda.is_available())

如果你安装了支持 CUDA 的 PyTorch，第二行代码应该返回 True，否则返回 False。

停止虚拟环境

当你完成了 PyTorch 的使用，你可以停止虚拟环境：

deactivate

这个命令将从虚拟环境中退出，恢复到原始系统环境。

这就是如何在虚拟环境中安装 PyTorch 的方法了。希望这篇文章可以帮助你开始使用 PyTorch。