win安装Pytorch
时间: 2025-05-07 09:05:08 浏览: 14
### PyTorch 在 Windows 系统上的安装方法
要在 Windows 上成功安装 PyTorch,可以按照以下方式进行操作:
#### 使用 Anaconda 安装 PyTorch
推荐通过 Anaconda 来管理 Python 和其依赖项,因为这种方式更加稳定且易于维护。
1. **创建并激活 Conda 虚拟环境**
首先需要创建一个新的虚拟环境来隔离 PyTorch 及其相关库的安装。可以通过以下命令实现:
```bash
conda create -n pytorch python=3.9
```
创建完成后,激活该虚拟环境:
```bash
conda activate pytorch
```
2. **安装 PyTorch GPU 或 CPU 版本**
如果计算机配备了 NVIDIA 显卡并且希望利用 CUDA 加速,则可以选择安装带有 GPU 支持的 PyTorch;如果不需要 GPU 支持或者设备不兼容 CUDA,则可选择仅安装 CPU 版本。
- 对于 GPU 版本(假设使用的 CUDA 工具包版本为 `cu118`),执行如下命令:
```bash
conda install pytorch torchvision torchaudio cudatoolkit=11.8 -c pytorch -c nvidia
```
此处指定了 `-c pytorch` 和 `-c nvidia` 渠道以确保获取最新稳定的构建[^2]。
- 若只需 CPU 版本,可以直接运行此命令:
```bash
conda install pytorch torchvision torchaudio cpuonly -c pytorch
```
3. **验证安装是否成功**
成功安装后,在 Python 解释器中测试导入模块的功能是否正常工作。
```python
import torch
print(torch.__version__)
print(torch.cuda.is_available()) # 检查是否有可用的CUDA支持
```
上述过程涵盖了从准备到最终确认的所有必要步骤[^3]。
---
阅读全文
相关推荐

















