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版本
查看虚拟环境内的cuda、cudnn版本
最新推荐文章于 2024-08-06 15:26:26 发布