torch安装

1、cuda

如何查看已安装的cuda版本？

在cmd中输入：nvcc --version

2、torch

torch下载链接：download.pytorch.org/whl/torch_stable.html

在里面根据自己的需要选取合适的torch下载即可，这里我选择的是这个：

torch比较大，大概2个多G

下载完成后，直接把它放入要安装的pycharm项目中，然后在pycharm终端使用：

pip install torch-2.0.0+cu117-cp38-cp38-win_amd64.whl命令安装即可

部分内容参考文章：cuda11.2版本的对应安装的pytorch版本_cuda11.2对应的pytorch-优快云博客

3、torchvision

torch和torchvision对应的版本参考这篇文章：torch torchvision版本对应关系_torch与torchvision版本对应-优快云博客

按照上面的torch，本文选择的torchvision版本如下：

下载完成之后，安装方法同torch：

pip install torchvision-0.15.0+cu117-cp38-cp38-win_amd64.whl