【torch-gpu】手动快速安装

cuda11.7就选cu117
python3.8就选cp38
windows就选win

安装torch

进入官网
https://download.pytorch.org/whl/torch/
搜索
torch-2.0.1+cu117-cp38-cp38-win_amd64.whl

安装torchvision

进入官网
https://download.pytorch.org/whl/torchvision/
搜索
torchvision-0.15.2+cu117-cp38-cp38-win_amd64.whl

安装torchaudio

进入官网
https://download.pytorch.org/whl/torchaudio/
搜索
torchaudio-2.0.2+cu117-cp38-cp38-win_amd64.whl

pip安装

进入下载目录
目录栏输入cmd回车
激活虚拟环境activate py38torch
开始安装
pip install torch-2.0.1+cu117-cp38-cp38-win_amd64.whl
pip install torchvision-0.15.2+cu117-cp38-cp38-win_amd64.whl
pip install torchaudio-2.0.2+cu117-cp38-cp38-win_amd64.whl

不要选错版本，不然会提示：
torchaudio 2.0.2+cu117 requires torch2.0.1, but you have torch 1.13.0+cu117 which is incompatible.
torchvision 0.15.2+cu117 requires torch2.0.1, but you have torch 1.13.0+cu117 which is incompatible .

验证GPU可用

import torch

# Check if GPU is available
if torch.cuda.is_available():
    print("GPU 可用")
else:
    print("GPU 不可用")