pytorch
文章平均质量分 76
横推八百无对手,轩辕重出武圣人
爱好广泛,涉猎广泛。
展开
-
Pip快速离线安装pytorch-gpu
Pip快速离线安装pytorch-gpu1小节讲解涉及的基本概念,读者也可直接从2小节开始。1 安装pythorch涉及的基本组件概念1.1 显卡驱动Driver常识概念,此处略过。1.2 CUDACUDA(Compute Unified Device Architecture)是显卡厂商NVIDIA推出的运算平台。CUDA™是一种由NVIDIA推出的通用并行计算架构,是一种并行计算平台和编程模型,该架构使GPU能够解决复杂的计算问题。运行CUDA应用程序需要要求系统具有一个具有CUDA功能的G原创 2021-09-10 20:07:13 · 3446 阅读 · 3 评论 -
torch.stack()按轴叠加原理
1. 问题:大部分帖子和一些典型的介绍numpy、pytorch的书籍对此部分并没有详细的介绍,仅仅简单地在np.cat()或torch.stack()等中提到当连接轴指定为0,或1时按照某某连接、排列。然而,当连接轴值较大时(如3,4,5),笔者发现,这些函数的输出并不是大多数材料所写的那样。笔者甚至看不出任何规律,为探寻其轴操作的原理,做了如下实验。2 归纳:定义两个初始矩阵:执行pytorch中的torch.stack([A, B], 2)操作,因为[A, B]可理解为将A,B连接,即t原创 2021-08-07 21:14:49 · 503 阅读 · 0 评论