win10 pytorch用GPU 运行代码

1:  首先安装显卡对应的CUDA ，

2：打开pytorch 官网

   

3：验证：

import torch
DEVICE = ("cuda" if torch.cuda.is_available() else "cpu")
print(torch.cuda.is_available())

输出为true 则证明 已经可以启动显卡运行程序