# 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))
#注意!安装目录要英文目录不要搞在中文目录 !不然可能报些奇奇怪怪的bug!
windows上报错:Could not find a version that satisfies the requirement torch==0.4.1->安装pytorch0.4.1 :https://blog.csdn.net/weixin_41010198/article/details/103107083
本站文章如无特殊说明,均为本站原创,如若转载,请注明出处:pytorch gpu~ cuda cudacnn安装是否成功的测试代码 - Python技术站