一、配置国内镜像源
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,进行安装即可。