简介:使用 Anaconda 安装 PyTorch
使用 Anaconda 安装 PyTorch
PyTorch 是由 Facebook 开发的广泛使用的深度学习框架。近年来,越来越多的研究者以及开发人员转向 PyTorch,因为其具有灵活性高,易于上手等特点。但是,其安装过程可能会让一些用户感到困扰。在这里,我们将详细介绍如何使用 Anaconda 来安装 PyTorch。
Anaconda 是一个流行的数据科学平台,它提供了大量的科学计算和数据分析工具包,包括 Python 语言本身。使用 Anaconda,可以方便地管理多个 Python 环境和相关的包。
以下是使用 Anaconda 安装 PyTorch 的步骤:
这里的 “myenv” 是你新环境的名称,”python=3.8” 是 Python 的版本。你可以根据需要更改这些值。
conda create -n myenv python=3.8
conda activate myenv
对于 CPU 版本的 PyTorch:
conda install pytorch torchvision torchaudio -c pytorch
这里的
conda install pytorch torchvision torchaudio -c pytorch_cpu
torchvision 和 torchaudio 是 PyTorch 的扩展包,分别用于计算机视觉和音频处理任务。如果你只需要 PyTorch 的基本功能,可以只安装 pytorch。如果一切正常,你应该能看到 PyTorch 的版本号被打印出来。
import torchprint(torch.__version__)
然后,你可以使用
conda deactivate
conda activate base 来重新激活 “base” 环境。