深度学习前置知识

• PyTorch 环境的安装

  • GPU：

    • 有 Nvidia 独立显卡

    • 显存 >= 4G

    • 推荐：12G 显存（性价比较高）

    • 显卡驱动正常（能够正常使用）

    • 官网：PyTorch

      • pip install torch torchvision --index-url https://download.pytorch.org/whl/cu121
      • conda install pytorch torchvision pytorch-cuda=12.1 -c pytorch -c nvidia

    • 代码下：

      • import torch
      • torch.cuda.is_available()

    • 命令行：

      • nvidia-smi

  • CPU：

    • pip install torch torchvision
    • conda install pytorch torchvision cpuonly -c pytorch

• 在线资源：

  • modescope.cn
    • 领取免费资源
  • virtaicloud.com
    • 领取免费资源