PyTorch 在 Google Colab 中的使用指南 📚

快速入门 🚀

1. 安装 PyTorch
   在 Colab 中可通过以下命令安装最新版本：

   !pip install torch torchvision torchaudio
   

   ✅ 安装完成后，可使用 import torch 验证是否成功。

2. 验证环境
   运行简单代码测试 GPU 支持：

   print(torch.cuda.is_available())  # 输出 True 表示成功启用 GPU
   

优质资源 🌐

• PyTorch 官方 Colab 示例（点击查看 GPU 加速演示）
• PyTorch 教程合集（涵盖基础到高级用法）
• Colab 环境配置指南（优化运行效率技巧）

图形化展示 🖼️

常见问题 ❓

• 如何持久化文件？ 使用 !wget 或 !curl 下载文件到 Colab，或通过 Google Drive 挂载。
• 内存不足怎么办？ 尝试关闭不必要的服务或使用免费版 Colab（默认 12GB 显存）。

✨ 提示：使用 torchvision 可直接加载预训练模型，例如：

from torchvision.models import resnet18
model = resnet18(pretrained=True)