一、确定电脑有英伟达显卡
点击任务控制台中的性能查看是否有显卡
二、查看CUDA Driver版本
nvidia-smi
三、使用pip安装Pytorch
- 进去Pytorch官网,点击Install
- 安装的CUDA Runtime版本应小于CUDA Driver版本
此处CUDA Runtime版本是11.7,此电脑的CUDA Driver版本是12.1 - 复制命令在命令行里执行
(1)如果执行出现如下错误
代表python版本不匹配,官网写的需要python3.7-python3.9
(2)安装成功
四、安装验证
输入以下命令,如果显示True,代表Pytorch安装成功
import torch
torch.cuda.is_available()