深度学习
撸码小白
爬虫开发,机器学习,深度学习
展开
-
Failed to get convolution algorithm. This is probably because cuDNN failed to initialize
最近在学tf2,训练模型的时候遇到以下错误: 解决方案如下(添加以下代码): physical_devices = tf.config.experimental.list_physical_devices('GPU') assert len(physical_devices) > 0, "Not enough GPU hardware devices available" tf.config.experimental.set_memory_growth(physical_devices[0], Tr原创 2020-12-28 18:25:48 · 217 阅读 · 0 评论 -
pytorch-自定义5×5卷积核对图片的模糊化操作
这里写目录标题1.1 导入类库1.2 图片处理1.3自定义卷积核1.4 模糊化操作1.5 维度转换1.6 图片保存1.7 最终结果展示(原始图片,5×5滤波器处理图片)①原始图片②经过5×5滤波器处理后的图片1.8 总结 1.1 导入类库 // An highlighted block import torch import matplotlib.pyplot as plt import numpy as np # 读取一张图片并进行展示 pic=plt.imread('./数据集/taijiling.j原创 2020-06-10 18:06:47 · 3346 阅读 · 1 评论