找到自己显卡版本的cuda下载对应的pytorch
参考:各版本pytorch
注意:
链接的cu代表cuda,cp代表python版本
同一个pytorch版本有不同的cuda版本,下好后可用torch.version.cuda查看当前pytorch是否与显卡cuda版本对应
匹配上后用torch.cuda.is_available()验证调用性
验证算力:
a=torch.Tensor([1,2])
a = a.cuda(0)
找到自己显卡版本的cuda下载对应的pytorch
参考:各版本pytorch
注意:
链接的cu代表cuda,cp代表python版本
同一个pytorch版本有不同的cuda版本,下好后可用torch.version.cuda查看当前pytorch是否与显卡cuda版本对应
匹配上后用torch.cuda.is_available()验证调用性
验证算力:
a=torch.Tensor([1,2])
a = a.cuda(0)