- 检查CUDA
- 检查cudnn
命令行终端启动python
检查CUDA>>> import torch >>> torch.cuda.is_available() # 检查cuda是否可用 >>> torch.version,cuda # 查看cuda版本检查cudnn
>>> torch.backends.cudnn.is_available() # 检查cudnn是否可用 >>> torch.backends.cudnn.version() # 查看cudnn版本
命令行终端启动python
检查CUDA>>> import torch >>> torch.cuda.is_available() # 检查cuda是否可用 >>> torch.version,cuda # 查看cuda版本检查cudnn
>>> torch.backends.cudnn.is_available() # 检查cudnn是否可用 >>> torch.backends.cudnn.version() # 查看cudnn版本
名师互学网 版权所有 (c)2021-2022 ICP备案号：晋ICP备2021003244-6号
