Windows安装Torch

Windows安装Torch

二级标题

查看CUDA版本信息

在window下，打开CMD界面，输入如下指令即可

nvidia-smi

在相应的输出中，可以知道CUDA的版本为12.0

安装Torch

一般情况下，我们需要安装的Torch组件包括三个模块需要安装，分别为：
Torch，TorchVision，TorchAudio

版本对应关系

Torch是主版本，因此，需要基于实际安装的torch，选择不同的TorchVision和TorchAudio版本。
Torch和TorchVision版本对应关系可见下表
Torch和TorchAudio的版本对应关系

Torch相应安装包的下载

Torch的相关安装包可以通过如下的官网链接进行相应的下载。
https://download.pytorch.org/whl/torch_stable.html
下载的注意事项：

• 版本的对应
  文件名的cpu或者cu110等，这些代表着这个是cpu还是gpu的，而cu110则代表gpu的cuda版本为11.0。
  cp代表Python版本，如cp310则代表Python3.10版本。
  win_amd和linux_x比较容易理解，代表系统版本。
  比如图示的代表是按爪给你GPU版本，CUDA的版本为11.6，Torch版本为1.13.1，Python版本为3.9，安装的是Windows平台。

• 版本兼容性
  在实际过程中，可能发现对应的CUDA没有相应的安装包版本对应，此时可以考虑前一个版本的对应安装包，因为CUDA是向下兼容的，比如，我们的CUDA版本为12.0，可以考虑安装CUDA为11.8的Torch。

离线安装

1. 在安装命令栏中，将文件定位到.whl文件所在未知，指令如下

cd path 

执行后，结果如下
2. 执行如下指令，系统将自动安装Torch安装包

pip install torch-2.2.2+cu118-cp310-cp310-win_amd64.whl

执行后，结果如下：

3. 执行如下指令，系统将自动安装TorchVison安装包

pip install torchaudio-2.2.2+cu118-cp310-cp310-win_amd64.whl

执行后，结果如下：

4. 执行如下指令，系统将自动安装TorchAudio安装包

pip install torchvision-0.17.2+cu118-cp310-cp310-win_amd64.whl

执行后，结果如下：

结果验证

在命令行中，输入如下指令：

import torch
print（torch.__version__int ）
print(torch.cuda.is_available())

输入结果如下即大功告成、

[1] 版本对应关系
[1] Torch离线下载