
PyTorch
文章平均质量分 56
ErbaoLiu
计算数学 本硕
展开
-
连接不上hugging face的解决方法
【代码】连接不上hugging face的解决方法。原创 2024-10-20 12:44:17 · 817 阅读 · 0 评论 -
PyTorch从零实现和可视化Transformer注意力(Additive Attention、Scaled Dot Product Attention、Multi-Head Attention)
上面公式没有体现批量操作,每个矩阵可以添加batch_size维度,例如Q的维度为(batch_size,n,d)。翻译 2024-10-16 10:55:15 · 360 阅读 · 0 评论 -
使用PyTorch从0实现Fashion-MNIST数据集分类
【代码】使用PyTorch从0实现Fashion-MNIST数据集分类。原创 2024-10-12 20:37:00 · 481 阅读 · 0 评论 -
Windows11安装CUDA、cuDNN、PyTorch详解
cuDNN(CUDA Deep Neural Network library)是一个由NVIDIA开发的深度学习GPU加速库,旨在为深度学习任务提供高效、标准化的原语(基本操作)来加速深度学习框架在NVIDIA GPU上的运算。此外,PyTorch还具有高度的灵活性和可扩展性,支持多种硬件平台,并且有一个活跃的社区,提供了大量的教程和资源,使得用户可以快速上手并解决实际问题。PyTorch的核心特点包括:12。可以看到NVIDIA GPU驱动版本为546.56,支持的CUDA最高版本为12.3。原创 2024-10-12 08:33:59 · 14550 阅读 · 10 评论