Windows下安装Pytorch-GPU

一、确定电脑有英伟达显卡

点击任务控制台中的性能查看是否有显卡
Windows下安装Pytorch-GPU_第1张图片

二、查看CUDA Driver版本

nvidia-smi

Windows下安装Pytorch-GPU_第2张图片

三、使用pip安装Pytorch

  1. 进去Pytorch官网,点击Install
    Windows下安装Pytorch-GPU_第3张图片
  2. 安装的CUDA Runtime版本应小于CUDA Driver版本
    Windows下安装Pytorch-GPU_第4张图片
    此处CUDA Runtime版本是11.7,此电脑的CUDA Driver版本是12.1
  3. 复制命令在命令行里执行
    (1)如果执行出现如下错误
    Windows下安装Pytorch-GPU_第5张图片
    Windows下安装Pytorch-GPU_第6张图片
    代表python版本不匹配,官网写的需要python3.7-python3.9
    (2)安装成功
    Windows下安装Pytorch-GPU_第7张图片

四、安装验证

输入以下命令,如果显示True,代表Pytorch安装成功

import torch
torch.cuda.is_available()

在这里插入图片描述

你可能感兴趣的:(环境安装,pytorch,windows,python)