文章目录
- 前言
- 尝试方法
- 一、尝试方法一
- 二、尝试方法二
- 总结
前言
之前用服务器跑的时候,发现是可以跑的。但当有其他人一同使用的时候,就会抛出:No CUDA GPUs are available,这个时候我尝试了以下两种方式解决,后面终于解决了
尝试方法
一、尝试方法一
import os
os.environ['CUDA_VISIBLE_DEVICES'] = "0"
当时能运行一段时间但之后显示被占用,于是用nvidia-smi查看内存占用情况,发现这个时间段用的人比较多
二、尝试方法二
import os
os.environ['CUDA_VISIBLE_DEVICES'] = "1"
把0改成1再次运行,就能正常运行了,看了看程序运行时间,跑了8个小时才跑完执行脚本
总结
除了使用os.environ[‘CUDA_VISIBLE_DEVICES’] = “1”,在此基础上也可以尝试调小代码中的patience、epochs值试下