Anaconda安装PyTorch教程

CPU（无GPU）版安装

1. Anaconda创建PyTorch环境

• 打开 Anaconda Powershell Prompt，输入

conda create -n (环境名称) python=3.7

example:
conda create -n pytorch-cpu python=3.7

其中python=3.7为指定该环境的python版本。

• 输入conda activate pythorch_cpu进入环境。

2. 安装PyTorch

进入PyTorch官网，根据据提示选择相应的PyTorch版本，会生成相应的命令，如图：

输入conda install pytorch torchvision cpuonly -c pytorch，根据提示输入y，等待片刻即可安装完成。

GPU版安装

1. Anaconda创建PyTorch环境

1. 打开 Anaconda Powershell Prompt，输入

conda create -n (环境名称) python=3.7

example:
conda create -n pytorch-gpu python=3.7

2. 输入conda activate pytorch-gpu进入环境。

2. 安装PyTorch

• cuda环境搭建
• 进入PyTorch官网安装页面，选择版本后会生成安装命令。
• 输入命令conda install pytorch torchvision cudatoolkit=10.2 -c pytorch，根据提示输入y，等待片刻即可安装完成。