2020年3月5日
pytorch gpu~ cuda cudacnn安装是否成功的测试代码
摘要： # CUDA TEST import torch x = torch.Tensor([1.0]) xx = x.cuda() print(xx) # CUDNN TEST from torch.backends import cudnn print(cudnn.is_acceptable(xx))# 阅读全文
posted @ 2020-03-05 00:02 冰河入梦~ 阅读(3408) 评论(0) 推荐(0) 编辑