常见问题通知
本页用于汇总同学们经常问的问题,将持续更新,欢迎随时反馈补充。
Q1:显存不够怎么办?如何切换显卡
如果遇到显存不够的情况,请按以下步骤切换到空闲显卡:
- 重启当前的 Kernel(Jupyter Lab/Notebook 菜单中选择
Kernel -> Restart)。 - 在命令行中执行
nvidia-smi查看每块显卡的显存占用,选择一块显存剩余充足的卡。假设选到的卡号为 2(第三张卡)。 - 在 Jupyter 的最顶部插入一个单元格,并在导入 PyTorch/深度学习库之前执行以下代码:
import os
os.environ['CUDA_VISIBLE_DEVICES'] = '2' # 注意:这里填写刚刚的卡号,不一定是 2
提示:
- 上述环境变量需在本次会话中最先设置,如已导入
torch则需先重启 Kernel 再设置。 - 该设置仅对当前 Kernel 生效;重启后需要重新设置。