《PyTorch面试精华》


1 前言
    1.1  PyTorch安装

    1.2  显卡驱动的困惑

    1.3  CUDA安装注意事项

    1.4  cuDNN的介绍

    1.5  Pytorch Lightning介绍

    1.6  PyTorch学习之道

    1.7  PyTorch快速入门

    1.8  PyTorch调参之道

    1.9  PyTorch调参套件

    1.10  手动创建虚拟环境

2 深度学习之数学基础
    2.1  希腊字母解读

    2.2  梯度的物理意义

    2.3  图解梯度下降法

    2.4  图解梯度上升法

    2.5  自然梯度

    2.6  泰勒公式的介绍

    2.7  信息与信息熵

    2.8  重要性采样

    2.10  欧几里得范数

    2.11  特征值和特征向量

    2.12  似然函数的理解

    2.13  矩阵秩的深刻理解

3 PyTorch入门疑难点
4 PyTorch全局设置
    4.1  全局设置当前设备

    4.2  全局设置浮点精度

5 PyTorch GPU分布式训练
    5.1  PyTorch GPU基础操作

    5.2  DataParallel用法详解

    5.3  GPU分布式训练模型

    5.4  CUDA_VISIBLE_DEVICES

    5.5  device详细说明

    5.6  to(device)和.cuda()

    5.7  CUDA设备索引

    5.8  GPU设备索引

6 向量的基础与核心
    6.1  Tensor的组成与存储

    6.2  Tensor的grad属性

    6.4  Tensor的叠加

    6.5  禁用梯度计算

    6.6  向量的保存和加载

    6.7  参数向量

    6.8  叶子节点

    6.9  detach原理

    6.10  requires_grad属性

    6.11  Tensor与Numpy互换

    6.12  张量cat操作

    6.13  零维张量

    6.15  squeeze/unsqu...函数

    6.16  argmax和max的区别

    6.17  torch.as_tensor的应用

7 神经网络基础
    7.2  PyTorch计算图

    7.3  查看网络权重参数

    7.4  保存模型

    7.5  Adam相关面试题

    7.6  Train模式和Eval模式

    7.7  线性网络

    7.8  双线性网络

    7.9  惰性线性层

    7.10  PyTorch中的自动微分

    7.12  Dropout机制

    7.13  半精度训练

    7.14  Xavier初始化

    7.15  注意力机制

    7.16  Dataset数据处理

    7.17  StepLR学习率调度器

    7.18  词嵌入的理解

    7.19  TensorDataset的使用

    7.20  模型的保存与加载

    7.21  ModuleList和Sequential

    7.22  Batch Normalization介绍

8 计算机视觉基础知识
    8.1  通道的深刻理解

    8.2  1x1卷积的作用

    8.3  特征提取和可视化

    8.4  反卷积的推导

    8.5  理解卷积

    8.7  空洞卷积

    8.8  池化层的作用

    8.9  感受野与特征图

    8.10  NMS算法

    8.11  特征图尺寸计算

9 循环神经网络基础
    9.2  RNN的介绍

10 注意力机制
    10.1  位置编码的作用

    10.2  位置编码的种类

    10.4  Embedding本质理解

    10.6  Transformer VS CNN/RNN

    10.7  ELMo介绍

11 PyTorch归一化
    11.2  层归一化技术详解

12 激活函数相关内容
    12.1  激活函数简介

    12.2  万能逼近定理

    12.3  指数函数的学习

    12.4  Sigmoid函数的介绍

    12.5  Tanh函数的介绍

    12.6  Softmax函数的实现

    12.7  ReLU函数的介绍

    12.8  Leaky Relu函数的介绍

    12.9  ReLu与非线性的理解

    12.10  Parametric ReLU函数

    12.11  ELU函数介绍

    12.12  神经元死亡的问题

13 思考题的答案
    13.1  思考题的答案解密

to(device)和.cuda()的区别

创建时间:2024-09-15 更新时间:2024-09-15 阅读次数:1824 次

1 .cuda()方法简介

在PyTorch中,即使是有GPU的机器,它也不会自动使用GPU,而是需要在程序中显示指定。当我们调用model.cuda(),PyTorch可以将模型加载到GPU上去,但是这种方法不被提倡,而建议使用model.to(device)的方式,这样可以显示指定需要使用的计算资源,特别是有多个GPU的情况下。

如果两个数据相加,例如a和b,要么在数据后面都加上cuda(),如a.cuda()和b.cuda(),使数据在GPU上进行运算;要么使用另外一种方式:在数据后面都不加cuda(),采用CPU运算。如果一个数据加了cuda()而另一个数据没有加,在运算的时候,会造成类型不匹配而报错。因为GPU版本是torch.cuda.FloatTensor,CPU版本是torch.FloatTensor。

2 .to(device)和.cuda()的区别

在PyTorch中,.cuda()方法和.to(device)方法都可以用来将模型或张量(Tensor)移动到不同的设备上进行计算。虽然这两种方法在某些情况下可以达到相似的效果,但它们之间存在一些关键的区别:

2.1 .cuda()方法介绍

  • 特定性:.cuda()方法专门用于将模型或张量移动到GPU上。它是一个快捷方法,专为CUDA设备设计。

  • 使用方式:当你调用.cuda()时,如果你的系统中有多个GPU,你可以通过传递一个设备索引来指定模型或张量应该被移动到哪个GPU上,比如.cuda(1)。如果不指定索引,则默认移动到系统的默认GPU。

  • 限制:.cuda()仅适用于CUDA兼容的GPU设备,不能用于将模型或张量移动到CPU或其他类型的设备。

2.2 .to(device)方法介绍

  • 通用性:.to(device)方法提供了一种更通用的方式来移动模型或张量,它允许你指定任何设备作为目标。这里的设备可以是CPU("cpu")、CUDA设备 ("cuda"或"cuda:0"等指定GPU索引的方式)、甚至是其他PyTorch支持的设备类型。

  • 灵活性:.to(device)方法让代码更具可移植性和灵活性。你可以通过更改device变量的值来控制模型或张量的移动,而无需更改方法调用或使用条件编码。示例:假设device = torch.device("cuda:0"),那么调用.to(device)会将模型或张量移动到第一个GPU上。如果device = torch.device("cpu"),相同的代码行将模型或张量移动回CPU。

2.3 个人总结

使用.cuda()当你确定目标设备是CUDA兼容的GPU,并且你不需要考虑将模型或张量移动到其他类型的设备。使用.to(device)来编写更通用、可适应不同设备的代码,特别是当你希望你的代码能够在没有GPU的环境中回退到CPU时,或者你需要处理多种设备类型时。因此,.to(device)方法提供了更高的灵活性和通用性,是推荐的方式来编写设备无关的PyTorch代码。

总之,大家要记住一点:.to(device)方法可以自由地将向量和模型在CPU和GPU等设备中迁移,这就是英文单词“to”的魅力所在吧。

本教程共117节,当前为第38节!
本教程最新修订时间为:2026-05-08 11:10:53