pytorch安装WIN
时间: 2025-01-14 13:56:09 浏览: 38
### 如何在 Windows 上安装 PyTorch
#### 安装前的准备工作
为了确保顺利安装 GPU 版本的 PyTorch,在开始之前需确认计算机已配备 NVIDIA 显卡并更新至最新的显卡驱动程序[^1]。如果右键菜单中未发现 NVIDIA 控制面板选项,则可能需要重新安装或修复显卡驱动程序以确保其正常工作[^2]。
#### 创建 Python 虚拟环境
建议先创建一个新的 Python 虚拟环境来隔离项目依赖项。可以使用 Anaconda 或者 Python 自带的方式建立虚拟环境:
- **Anaconda 方式**
```bash
conda create -n pytorch_env python=3.9
conda activate pytorch_env
```
- **Python 原生方式**
```bash
python -m venv pytorch_env
.\pytorch_env\Scripts\activate.bat
```
#### 下载 CUDA 和 cuDNN
对于希望利用 GPU 加速计算的应用场景,还需要预先下载与所选 PyTorch 版本兼容的 CUDA Toolkit 和 cuDNN 库文件,并按照官方指南完成相应设置[^5]。
#### 安装 PyTorch
通过 `pip` 工具可以直接在线获取最新发布的 PyTorch 包;而对于网络受限的情况则可以选择离线安装方案。以下是两种常见的安装途径:
- **在线安装**
对于大多数用户而言,最简便的方法是从官方网站获取推荐命令行来进行快速部署:
```bash
pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu118
```
- **离线安装**
当处于无法访问互联网环境中时,可提前准备好对应版本的.whl 文件后执行如下指令实现本地化加载:
```bash
pip install path_to_your_downloaded_file.whl
```
#### 验证安装成果
最后一步是在激活后的虚拟环境下运行简单的测试脚本来检验是否成功启用了 GPU 支持功能:
```python
import torch
print(torch.cuda.is_available())
```
当上述代码返回 True 即表明已经正确设置了基于 CUDA 的 PyTorch 运行环境[^4]。
阅读全文
相关推荐


















