在这个教程中,我们将向您展示如何在 Python 中安装 PyTorch。PyTorch 是一个流行的开源机器学习库,广泛用于深度学习。
安装步骤
检查 Python 版本 确保您的 Python 版本至少为 3.6。您可以使用以下命令检查 Python 版本:
python --version
检查 Python 版本创建虚拟环境 创建一个虚拟环境可以避免与系统中的其他项目发生冲突。您可以使用以下命令创建虚拟环境:
python -m venv myenv
创建虚拟环境激活虚拟环境 在 Windows 上,您可以使用以下命令激活虚拟环境:
myenv\Scripts\activate
在 macOS/Linux 上,您可以使用以下命令激活虚拟环境:
source myenv/bin/activate
激活虚拟环境安装 PyTorch 您可以使用以下命令安装 PyTorch:
pip install torch torchvision torchaudio
安装 PyTorch验证安装 安装完成后,您可以使用以下命令验证 PyTorch 是否已正确安装:
import torch print(torch.__version__)
验证 PyTorch 安装
下一步
现在您已经成功安装了 PyTorch,您可以开始构建自己的深度学习模型了。如果您需要更多帮助,请访问我们的 深度学习教程。