驗證cuda 及cudnn是否成功安裝

import torch
from torch.backends import cudnn

if __name__== "__main__":
     print(torch.cuda.is_available())
     x =  torch.tensor([10.])
     x = x.cuda()
     print(cudnn.is_acceptable(x)) 

torch.backends.cudnn.enabled = True#啟動cudnn
torch.backends.cudnn.benchmark = True#啟動cudnn的不確定性算法的自動搜索優化模式，適合輸入數據維度比較固定的情況，否則應該關閉此項。