weixin_45014813的博客

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的维度。

注：不明白为什么不match,代码仅供示例。

【代码】cuda学习之路--04--确定最优设备。

【代码】cuda学习之路--03--在cuda上跑一个二维网格和二维块的矩阵加法运算。

【代码】cuda学习之路--02--在cuda上跑一个加法。

【代码】cuda学习之路--01--在cuda上跑一个hello,world!

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