
cuda
盼盼呀咦呀咦呀~
这个作者很懒,什么都没留下…
展开
-
【无标题】
3:PyTorch可以通过set_default_tensor_type函数设置默认使用的Tensor类型torch.set_default_tensor_type(‘torch.DoubleTensor’),在局部使用完后如果需要其他类型,则还需要重新设置回所需的类型。注:1:pytorch中默认的的数据类型是torch.FloatTensor,即torch.Tensor等同于torch.FloatTensor;2:16半精度浮点是专门为GPU上运行设置的;处理size为1的维度。原创 2025-01-21 16:08:55 · 210 阅读 · 0 评论 -
cuda学习之路--05--在cuda做矩阵转置
注:不明白为什么不match,代码仅供示例。原创 2024-08-07 15:54:55 · 318 阅读 · 0 评论 -
cuda学习之路--04--确定最优设备
【代码】cuda学习之路--04--确定最优设备。原创 2024-08-05 14:58:36 · 206 阅读 · 0 评论 -
cuda学习之路--03--在cuda上跑一个二维网格和二维块的矩阵加法运算
【代码】cuda学习之路--03--在cuda上跑一个二维网格和二维块的矩阵加法运算。原创 2024-08-05 14:40:44 · 197 阅读 · 0 评论 -
cuda学习之路--02--在cuda上跑一个加法
【代码】cuda学习之路--02--在cuda上跑一个加法。原创 2024-08-02 11:18:39 · 183 阅读 · 0 评论 -
cuda学习之路--01--在cuda上跑一个hello,world!
【代码】cuda学习之路--01--在cuda上跑一个hello,world!原创 2024-08-02 11:10:14 · 166 阅读 · 0 评论 -
cpu gpu cuda cudnn区别以及作用
cpu gpu cuda cudnnCPU适合串行计算,擅长逻辑控制。GPU擅长并行高强度并行计算,适用于AI算法的训练学习CUDA 是NVIDIA专门负责管理分配运算单元的框架cuDNN是用于深层神经网络的gpu加速库查看cuda 版本nvcc -Vcat /usr/local/cuda/version.txt查看cudnn 版本cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2查看Nvidia GPU版本nvidi原创 2021-08-30 15:08:49 · 518 阅读 · 0 评论