cpu、gpu版本torch安装

1.安装cpu版torch：

方式1：直接在PyCharm中点击file>setting>project>project interpreter，选择“+”，添加torch，不建议用这种方式安装，可能安装不成功。

方式2：在命令窗口输入pip install torch,默认安装cpu版本。

2. 安装GPU版torch

搞深度学习之类的，大家肯定都是安装GPU版torch,适合并行运算。
方式1：去官网（https://pytorch.org/get-started/locally/）选择对应版本，复制最底下的命令进行安装。

pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121

方式2：
第一步：cmd命令行下输入‘nvidia-smi’，查看本机cuda版本，

第二步：可以参考网上找的这张图，确定需要安装的torch版本。

第三步，去该官网https://download.pytorch.org/whl/torch_stable.html下载对应版本的whl文件，然后采用以下命令进行安装：

选择合适版本，下载后得到whl文件如torch-1.9.1+cu102-cp36-cp36m-win_amd64.whl，利用以下命令安装：
pip install torch-1.9.1+cu102-cp36-cp36m-win_amd64.whl

3.其他gpu版torch安装教程参考：

https://blog.csdn.net/qq_60143666/article/details/125644741