pytorch下驗證是否安裝了cuda和cuDNN

進入python環境，然後輸入以下代碼測試：

#判斷是否安裝了cuda
import torch
print(torch.cuda.is_available())  #返回True則說明已經安裝了cuda
#判斷是否安裝了cuDNN
from torch.backends import  cudnn 
print(cudnn.is_available())  #返回True則說明已經安裝了cuDNN