深度学习
文章平均质量分 85
MK422
这个作者很懒,什么都没留下…
展开
-
win10系统numba.cuda报错
from numba import cudaTPB = 16A = np.full((TPB * 13, TPB * 13), 3, np.float)B = np.full((TPB * 13, TPB * 13), 4, np.float)C_cpu = np.full((A.shape[0], B.shape[1]), 0, np.float)# start in GPU# 拷贝数据到设备端A_global_mem = cuda.to_device(A)B_global_me.原创 2020-11-03 19:02:04 · 1743 阅读 · 3 评论 -
win10安装CUDA以及GPU版的pytorch
1、安装NVIDIA显卡驱动(1)首先确定自己是否有NVIDIA显卡,如果没有可以关页面了;(2)一般装了系统都有NVIDIA显卡驱动;如果没有,可以点击下面的下载地址下载地址:http://www.nvidia.cn/Download/index.aspx?lang=cn,注意笔记本一般选择notebooks2、安装CUDA(1)确定显卡支持的CUDA版本NVIDIA控制面板—帮助—系统信息—组件,就可以看到显卡支持的CUDA版本,具体操作见下图:(2)根据自己显卡支持的C原创 2020-10-30 11:30:12 · 1767 阅读 · 0 评论 -
keras绘制网络模型图plot_model
1.绘制网络模型图方法from keras.models import Sequentialfrom keras.utils import plot_modelmodel = Sequential()构建网络编译网络训练模型评估网络预测模型plot_model(model, to_file='E://temp.png')2.遇到的问题(1)无法直接使用plot_model(),需要安装pydot,GraphViz解决方案:pip install pydot..翻译 2020-09-18 16:23:48 · 1302 阅读 · 0 评论