跑代码时出现这个错误,输入以下命令

$ nvidia-smi -l

解决RuntimeError: CUDA error: out of memory

原来是卡0内存爆了,于是指定相对空闲的2卡

CUDA_VISIBLE_DEVICES=2 python my_script.py

可以正常运行

相关文章:

  • 2022-12-23
  • 2021-08-09
  • 2022-12-23
  • 1970-01-01
  • 2021-05-09
  • 2022-12-23
  • 2021-11-23
  • 2022-12-23
猜你喜欢
  • 2021-10-07
  • 2021-06-17
  • 2021-10-30
  • 2021-11-22
  • 2021-05-24
  • 2022-01-14
  • 2021-06-16
相关资源
相似解决方案