pip安装Pytorch gpu版

Pytorch GPU 版安装

由于使用官方 pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu117 速度太慢，使用此方法下载资源，进行本地离线安装。

GPU准备

1. 查看显卡信息：
   Nvidia控制面板（右下任务栏通知）——系统信息（左下角）——组件——CUDA版本

2. 下载安装CUDA：
   根据1的CUDA版本英伟达官网下载：详尽参考

Pytorch 安装

需要安装的模块3个：torch 、torchvision和torchaudio。

新增最简单的方法

以下是cuda12.4环境下的安装方法，其他版本修改索引地址的cu124即可。 编辑于2025-01-28

pip install --upgrade torch torchvision torchaudio  --extra-index-url https://download.pytorch.org/whl/cu124

下载文件

1. 复制Run this Command:中的 https://download.pytorch.org/whl/cu117；

2. 访问网址，如图结果：
   

3. 下载torch，点击torch – 查找Ctr F – 内容cu117-cp310 – 右键单击torch-1.13.0+cu117-cp310-cp310-win_amd64.whl 目标资源 —复制链接–打开迅雷下载。（注：cu117–cuda11.7；cp310–python3.10；win–windows）

4. 下载torchvision，同上3，torchvision-0.14.0+cu117-cp310-cp310-win_amd64.whl
   5.下载torchaudio，同上3， torchaudio-0.13.0+cu117-cp310-cp310-win_amd64.whl

本地安装

在安装包目录下右键选择在 终端中打开，执行如下命令：

python -m pip install torch-1.13.0+cu117-cp310-cp310-win_amd64.whl
python -m pip install torchvision-0.14.0+cu117-cp310-cp310-win_amd64.whl
python -m pip install torchaudio-0.13.0+cu117-cp310-cp310-win_amd64.whl

torch，torchvision，torchaudio三者版本对应关系：参阅