# 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