首页pytorch-gpu下载pip

pytorch-gpu下载pip

时间: 2025-03-23 14:06:17 浏览: 34

### 如何使用 pip 安装支持 GPU 的 PyTorch 为了安装支持 GPU 的 PyTorch，可以按照以下方法操作： #### Python 版本确认确保所使用的 Python 版本在受支持范围内。根据官方文档说明，当前仅支持 Python 3.8 至 3.11 版本[^1]。 #### 使用 Pip 命令安装对于 Windows 平台上的 GPU 支持版本，可以通过如下命令完成安装： ```bash pip install torch==1.13.1+cu117 torchvision==0.14.1+cu117 torchaudio==0.13.1 --extra-index-url https://download.pytorch.org/whl/cu117 ``` 此命令会安装指定 CUDA 版本 (cu117 对应于 CUDA 11.7) 的 PyTorch 及其相关依赖项[^2]。 #### 验证安装是否成功进入 Python 解释器并运行以下代码来验证 GPU 是否可用以及安装是否正确： ```python import torch print(torch.cuda.is_available()) ``` 如果返回 `True`，则表明 GPU 加速功能已被启用，并且安装过程顺利完成[^3]。 #### Anaconda 环境下的额外步骤当在 Anaconda 中执行相同任务时，建议先创建一个新的虚拟环境以隔离不同项目间的依赖关系冲突。具体流程如下所示： ```bash conda create --name envname python=3.8 -y conda activate envname pip install torch==1.13.1+cu117 torchvision==0.14.1+cu117 torchaudio==0.13.1 --extra-index-url https://download.pytorch.org/whl/cu117 ``` 之后可通过 `conda list` 或者 `conda env list` 来检查新环境中已有的软件包列表及其状态信息[^4]。

阅读全文