一、安装准备
在安装PyTorch之前,请确保您的计算机满足以下要求:
- 操作系统:本报告以Windows操作系统为例,但PyTorch也支持其他操作系统,如Linux和macOS。
- Python版本:PyTorch支持Python 3.x版本。请确保您的Python版本与之兼容。
- NVIDIA显卡:如果您希望使用GPU进行加速计算,那么您的计算机需要配备支持CUDA的NVIDIA显卡。
- CUDA版本:根据NVIDIA显卡的型号和驱动版本,选择相应的CUDA版本。本报告以CUDA 11.4为例。
二、安装过程
-
下载Anaconda:首先,从Anaconda官网下载并安装Anaconda。Anaconda是一个流行的Python数据科学平台,它提供了大量的科学计算包和工具。
-
创建Conda环境:打开Anaconda Prompt(或终端),使用以下命令创建一个新的Conda环境(如果尚未创建):
conda create -n pytorch_env python=3.8 conda activate pytorch_env
这将创建一个名为
pytorch_env
的新环境,并激活它。 -
安装PyTorch:访问PyTorch官网,点击“Install”