gpu版pytorch安装

pytorch

pytorch官网：https://pytorch.org/get-started/locally/

目前gpu版pytorch安装不需要自己安装cuda，而且注意windows版仅支持cuda11.3 ，在镜像网站可以找到cuda11.1、11.3、11.5的版本
在这里网络环境好的可以直接使用命令安装，这里是conda安装方式

conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch

网络环境不好最好还是用镜像网站提供的文件，比如
pytorch：https://download.pytorch.org/whl/torch/
torchversion：https://download.pytorch.org/whl/torchvision/（这个装不装都行）
文件命名就是版本信息，比如：torch-1.11.0+cu115-cp38-cp38-win_amd64.whl，1.11.0是pytorch的版本号࿰