在深度学习过程中常涉及的命令记录备查。
本文中涉及命令均在windows上,使用Anaconda管理环境的情况下。
1.pytorch下查看cuda版本,查看cudnn版本
import torch
print(torch.version.cuda)
print(torch.backends.cudnn.version())
2.还有一种方式是在Anaconda里查看。但是由于创建配置环境时操作不同,导致查看不到。因此建议采用命令行的形式最佳。
下图是配置的tensorflow环境中:
下图是配置的pytorch环境中: