测试pytorc+cuda+cuDNN,pytorch的Gpu版本安装是否成功

import torch print(torch.cuda.is_available()) 返回结果是True,则PyTorch的GPU安装成功web
相关文章
相关标签/搜索