conda activate cp36
python

import torch

torch.cuda.is_available()   # cuda是否可用
torch.version.cuda          # cuda版本

torch.backends.cudnn.is_available()  # cudnn是否可用
torch.backends.cudnn.version()       # cudnn版本