Pytorch-gpu安装教程

这是参考博文点击进入

下载想要的CUDA,原来下载的是CUDA11.2，但是发现pytorch不支持CUDA11.2.于是卸载掉CUDA11.2，如果不知道如何卸载请自行百度，重新下载CUDA11.3.高版本的CUDA支持向下兼容，举个例子也就是说算力7.0的显卡可以在支持最高算力7.5的CUDA版本下运行，但是算力7.5的显卡不可以在支持最高算力7.0的CUDA版本下运行，下载并安装好对应的CUDA和cudnn后就去网站下载好对应的pytorch的whl文件，网址在这点击进入，这是我选择下载的，我cu113是CUDA11.3,cp38是支持python3.8

下载到自己想要安装的环境的对应的Scripts文件夹下

，然后打开cmd进入这个文件夹，激活这个环境，然后输入pip install whell

之后就可以进行安装了，输入pip install torch-1.10.0+cu113-cp38-cp38-win_amd64.whl

安装成功