Skip to content
GitHub

常见问题通知(持续更新)

常见问题通知

本页用于汇总同学们经常问的问题,将持续更新,欢迎随时反馈补充。

Q1:显存不够怎么办?如何切换显卡

如果遇到显存不够的情况,请按以下步骤切换到空闲显卡:

  1. 重启当前的 Kernel(Jupyter Lab/Notebook 菜单中选择 Kernel -> Restart)。
  2. 在命令行中执行 nvidia-smi 查看每块显卡的显存占用,选择一块显存剩余充足的卡。假设选到的卡号为 2(第三张卡)。
  3. 在 Jupyter 的最顶部插入一个单元格,并在导入 PyTorch/深度学习库之前执行以下代码:
import os
os.environ['CUDA_VISIBLE_DEVICES'] = '2'  # 注意:这里填写刚刚的卡号,不一定是 2

提示:

  • 上述环境变量需在本次会话中最先设置,如已导入 torch 则需先重启 Kernel 再设置。
  • 该设置仅对当前 Kernel 生效;重启后需要重新设置。