1.进入pytorch官网:https://pytorch.org
2.往下滑:看到配置信息(网页根据你的配置,自动帮你生成适合的安装指令)
3.查看自己电脑的配置信息。
a. 打开命令框,输入:nvidia-smi
b.然后根据生成的指令,使用pip指令安装即可
4.查看是否安装成功(运行一下python文件):
import torch device = 'cuda' if torch.cuda.is_available() else 'cpu' print(device)
结果显示:表示成功!
否则显示 :cpu