pytorch 安装

1. 在torch安装之前查看自己显卡的驱动版本
nvidia-smi
2.在官网选择适合自己环境的版本：

就是上面这张图的样子，系统、语言、版本啥的根据自己环境选择。然后复制下面生成的命令进行安装。

也可以复制生成的下载地址：

然后再浏览器打开：https://download.pytorch.org/whl/cu118
然后需要下载下面红框中的三个分别进行下载：

我的python版本是3.10，torch需要2.1版本，系统是win10 x64。
torch选择：torch-2.1.0+cu118-cp310-cp310-win_amd64.whl
（如果需要显卡驱动一定要选择CUXXX，如果没有显卡选择cpu）

torchaudio选择：torchaudio-2.1.0+cu118-cp310-cp310-win_amd64.whl

torchversion选择：torchvision-0.16.0+cu118-cp310-cp310-win_amd64.whl

这单个文件下载完成在进行命令安装(./ 为文件存在在当前项目路径)
pip install ./torch-2.1.0+cu118-cp310-cp310-win_amd64.whl
pip install ./torchaudio-2.1.0+cu118-cp310-cp310-win_amd64.whl
pip install ./torchvision-0.16.0+cu118-cp310-cp310-win_amd64.whl

最后通过一下命令验证：

import torch
torch.cuda.device_count()