windows下安装PyTorch框架支持CPU和GPU两种安装方式

在我之前的项目中用到PyTorch的场景并不多，所以自己对于这个深度学习框架的熟练度也不行，最近因为频繁需要用着PyTorch框架来做调试，需要自己安装搭建这个框架，之前只是用来做推理的，所以直接装的CPU，后面又搞来了GPU的环境这个也需要基于GPU来搭建PyTorch框架，这里主要来总结一下安装方法。

因为之前使用tensorflow和keras的时候直接库名称加版本号就可以了，但是pytorch这个稍微麻烦一点，所以想着专门汇总总结记录一下。

在我之前的文章里面写到过一篇相关的文章，不过当时做的比较简单，如下，感兴趣的话可以自行移步阅读：

《windows下安装Pytorch环境》

PyTorch官网在这里，首页截图如下：

 安装方式的话，PyTorch这里官方提供的是组合式选择式的命令生成方式，如下：

 针对不同的CUDA会有不同的安装命令。

下面主要是针对windows平台下GPU和CPU两种模式来进行安装命令汇总如下：

CUDA11.6

pip安装方式：
pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu116

conda安装方式：
conda install pytorch torchvision torchaudio pytorch-cuda=11.6 -c pytorch -c nvidia

CUDA11.7

pip安装方式：
pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu117

conda安装方式：
conda install pytorch torchvision torchaudio pytorch-cuda=11.7 -c pytorch -c nvidia

CPU

pip安装方式：
pip3 install torch torchvision torchaudio

conda安装方式：
conda install pytorch torchvision torchaudio cpuonly -c pytorch

接下来针对Linux平台来总结汇总GPU和CPU两种模式下的安装命令：

CPU

pip安装方式：
pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cpu

conda安装方式：
conda install pytorch torchvision torchaudio cpuonly -c pytorch

CUDA11.6

pip安装方式：
pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu116

conda安装方式：
conda install pytorch torchvision torchaudio pytorch-cuda=11.6 -c pytorch -c nvidia

CUDA11.7

pip安装方式：
pip3 install torch torchvision torchaudio

conda安装方式：
conda install pytorch torchvision torchaudio pytorch-cuda=11.7 -c pytorch -c nvidia

ROCm

pip安装方式：
pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/rocm5.2

conda安装方式：
暂不支持

如果是想要使用之前的版本可以参考这里，如下：

 记录备忘！