pytorch安装(windows)
时间: 2023-11-13 21:57:30 浏览: 66
pytorch安装介绍
好的,以下是在 Windows 上安装 PyTorch 的步骤:
1. 首先,确保你已经安装了 Python。如果没有安装,可以从官网下载并安装:https://www.python.org/downloads/windows/
2. 接着,打开命令提示符或者 Anaconda Prompt,输入以下命令安装 PyTorch:
```
pip install torch torchvision torchaudio
```
如果你的电脑上有 NVIDIA 显卡,可以安装支持 GPU 的版本:
```
pip install torch torchvision torchaudio -f https://download.pytorch.org/whl/cu111/torch_stable.html
```
3. 安装完成后,可以在 Python 中导入 PyTorch 模块进行测试:
```
import torch
print(torch.__version__)
```
如果输出了 PyTorch 的版本号,则说明安装成功。
阅读全文