windows下的pytorch-gpu环境搭建方法

一、配置国内镜像源

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/

#必不可少，不然会出现找不到包，或者只有cpu版本的pytorch包
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/ 
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/  
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/peterjc123/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
conda config --set show_channel_urls yes

1、访问官网：https://pytorch.org/，选择对应的命令进行安装；

2、通过conda install pytorch torchvision torchaudio cudatoolkit=10.2进行安装，建议去除官网中-c pytorch(代表通过官网下载，速度较慢)

二、检查是否安装成功

import torch

#检查GPU设备是否可用
torch.cuda.is_available()

#检查可用GPU设备数量
torch.cuda.device_count()

#获取GPU设备，设备索引默认从0开始
torch.cuda.get_device_name(0)

#检查当前设备索引
torch.cuda.current_device()

三、常见报错及解决

可能会出现如下错误：Error loading  "xxxx.dll"，是因为缺少visusal c++

下载https://aka.ms/vs/16/release/vc_redist.x64.exe，进行安装即可。