測試pytorc+cuda+cuDNN,pytorch的Gpu版本安裝是否成功

import torch print(torch.cuda.is_available()) 返回結果是True,則PyTorch的GPU安裝成功web
相關文章
相關標籤/搜索