PyTorch 是一个流行的开源机器学习库,适用于 Python 程序。以下是 PyTorch 在 Python 环境下的安装步骤。
安装步骤
安装 Python:首先确保你的系统中已经安装了 Python。推荐使用 Python 3.6 或更高版本。
创建虚拟环境:为了更好地管理项目依赖,建议创建一个虚拟环境。
python -m venv myenv source myenv/bin/activate # 在 Windows 上使用 myenv\Scripts\activate
安装 PyTorch:使用以下命令安装 PyTorch。
pip install torch torchvision torchaudio
你也可以使用以下命令安装 CPU 版本的 PyTorch。
pip install torch torchvision torchaudio cpuonly
验证安装:安装完成后,你可以通过以下命令验证 PyTorch 是否已正确安装。
import torch print(torch.__version__)
其他资源
如果你需要更多关于 PyTorch 的信息,可以访问我们的PyTorch 教程页面。
示例图片
以下是一些 PyTorch 相关的图片示例: