深度学习代码块之 忽略警告代码块 和 测试cuda和cudnn成功安装代码块

引言

在这篇博客中毛毛张将记录在深度学习过程中的一些代码块

忽略警告代码块

在编写深度学习代码的过程中，会出现一些警告，本来代码的报错已经够让人心烦的了，还出来个红色的警告更让人心烦，下面毛毛张将分享一个取消全文警告输出的代码块：

import warnings
warnings.filterwarnings("ignore")

加入这条语句之后的警告都将会被忽略，而之前的不受影响，因此如果想屏蔽掉所有的警告，可以放在代码的最前面。

测试cuda和cudnn成功安装代码块

import torch
print('torch版本：'+torch.__version__)
print('cuda是否可用：'+str(torch.cuda.is_available()))
print('cuda版本：'+str(torch.version.cuda))
print('cuda数量:'+str(torch.cuda.device_count()))
print('GPU名称：'+str(torch.cuda.get_device_name()))

参考文献

• https://blog.csdn.net/time_forgotten/article/details/104792200