首页怎么安装 CUDA 12.1 版本的 PyTorch

怎么安装 CUDA 12.1 版本的 PyTorch

时间: 2025-03-04 21:21:41 浏览: 388

### 安装带有特定 CUDA 版本的 PyTorch 为了安装支持 CUDA 12.1 的 PyTorch，可以按照官方推荐的方法通过 `pip` 或者 Anaconda 来完成。以下是基于 pip 的安装指南： #### 使用 Pip 安装 PyTorch 和 CUDA 12.1 支持确保已经正确设置了 CUDA 环境变量，并验证当前使用的 CUDA 版本是否为期望版本[^3]。 ```bash nvcc --version ``` 接着可以通过以下命令来安装兼容 CUDA 12.1 的 PyTorch 版本: ```python pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu121 ``` 这条指令会从指定 URL 下载并安装适用于 CUDA 12.1 的 PyTorch 软件包及其依赖项。对于那些偏好使用 Conda 进行包管理的人来说，则可以选择如下方式来进行安装： ```bash conda install pytorch torchvision torchaudio cudatoolkit=12.1 -c pytorch -c nvidia ``` 此命令会在 conda 渠道中寻找与 CUDA 12.1 对应的 PyTorch 发布版进行安装。 #### 验证安装成功与否一旦完成了上述任一过程之后，建议运行简单的测试脚本来确认 GPU 是否被有效识别以及能否正常工作： ```python import torch print(f"CUDA Available: {torch.cuda.is_available()}") print(f"Current Device: {torch.cuda.current_device()}") print(f"Torch Version : {torch.__version__}") ``` 如果一切顺利的话，这段代码应该能够显示出正确的 Torch 版本号，并表明 CUDA 是可用状态下的设备编号。

阅读全文