Appearance
PyTorch 配置
请先参考 Anaconda 配置 安装 Anaconda。
版本要求
PyTorch 需要满足以下版本要求：
| PyTorch 版本 | Python | Stable CUDA | Experimental CUDA |
|---|---|---|---|
| 2.0 | >=3.8, <=3.11 | CUDA 11.7, CUDNN 8.5.0.96 | CUDA 11.8, CUDNN 8.7.0.84 |
| 1.13 | >=3.7, <=3.10 | CUDA 11.6, CUDNN 8.3.2.44 | CUDA 11.7, CUDNN 8.5.0.96 |
| 1.12 | >=3.7, <=3.10 | CUDA 11.3, CUDNN 8.3.2.44 | CUDA 11.6, CUDNN 8.3.2.44 |
本文以 PyTorch 2.0、Python 3.10、CUDA 11.7 为例。
创建环境
在 PowerShell 中执行以下命令创建一个名为 pytorch、Python 版本为 3.10 的环境：
powershell
conda create -n pytorch python=3.10出现以下提示时，输入 y 确认安装：
powershell
Proceed ([y]/n)?激活环境：
powershell
conda activate pytorch安装 PyTorch
在 PowerShell 中执行以下命令安装 PyTorch：
powershell
conda activate pytorch
conda install pytorch torchvision torchaudio pytorch-cuda=11.7 -c pytorch -c nvidia出现以下提示时，输入 y 确认安装：
powershell
Proceed ([y]/n)?配置 Jupyter Notebook
请参考 VS Code Jupyter Notebook 配置 配置 Jupyter Notebook。