PytorchGPU版本

要求Driver Version>451.67

其中Driver Version是GPU 驱动版本型号，CUDA Version是CUDA 驱动 API 的型号。

下载CUDA10.2

检查安装是否成功：

nvcc -V

 CUDNN v8.3.1 

下载需首先注册会员

解压后重命名后移动到当前文件夹

打开anaconda prompt,为pytorch创建一个虚拟环境，与base等区分开.

  测试显卡是否可用

import torch
torch.cuda.is_available()
ngpu= 1
device = torch.device("cuda:0" if (torch.cuda.is_available() and ngpu > 0) else "cpu")
device
torch.cuda.get_device_name(0)
torch.rand(3,3).cuda()

True
cuda:0
GeForce MX250
tensor([[0.0745, 0.6669, 0.2551],
        [0.3763, 0.1458, 0.3693],
        [0.6710, 0.3850, 0.4676]], device='cuda:0')