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

被折叠的 条评论
为什么被折叠?



