pytorch怎么安装cuda
时间: 2025-01-06 20:33:44 浏览: 40
### 如何在 PyTorch 中安装配置 CUDA 支持
为了使 PyTorch 能够利用 NVIDIA 的 GPU 加速计算,需要确保系统已经正确安装了 CUDA 工具包。通常情况下,CUDA 会被默认安装到 `C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v{版本号}` 文件夹下[^1]。
对于 PyTorch 来说,推荐的方式是通过 pip 安装预编译好的二进制文件来获得带有 CUDA 支持的 PyTorch 版本。例如,可以使用如下命令安装特定 CUDA 和 Python 版本兼容的 PyTorch 及其配套工具:
```bash
pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu121
```
上述命令适用于 Windows 平台上的 AMD64 架构计算机,并且针对的是 CUDA 12.1 和 Python 3.12 这样的环境组合[^2]。请注意调整 URL 后缀中的 cuX 表示对应的 CUDA 版本以及确保所选轮子文件与本地系统的 Python 解释器相匹配。
如果想要验证安装是否成功并确认当前使用的 PyTorch 是否能够检测到可用的 GPU 设备,可以通过下面这段简单的测试代码来进行检查:
```python
import torch
if torch.cuda.is_available():
print(f"CUDA is available, version {torch.version.cuda}")
else:
print("CUDA not found.")
print(torch.__version__)
```
阅读全文
相关推荐

















