pytorch环境搭建
时间: 2023-10-18 21:31:42 浏览: 52
好的,以下是在 Windows 系统上搭建 Pytorch 环境的步骤:
1. 安装 Anaconda:首先需要下载并安装 Anaconda,下载地址为 https://www.anaconda.com/products/distribution。
2. 创建虚拟环境:打开 Anaconda Prompt,输入以下命令创建一个名为 pytorch 的虚拟环境。
```
conda create --name pytorch python=3.8
```
3. 激活虚拟环境:输入以下命令激活 pytorch 环境。
```
conda activate pytorch
```
4. 安装 Pytorch:根据自己的环境选择相应的 Pytorch 版本并安装。
- CPU 版本:
```
conda install pytorch torchvision torchaudio cpuonly -c pytorch
```
- GPU 版本:
```
conda install pytorch torchvision torchaudio cudatoolkit=11.1 -c pytorch -c nvidia
```
5. 验证安装:输入以下命令,如果没有报错则说明 Pytorch 安装成功。
```
python -c "import torch; print(torch.__version__)"
```
希望这些步骤可以帮助到你。