前言

之前用服务器跑的时候,发现是可以跑的。但当有其他人一同使用的时候,就会抛出: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值试下

Logo

鸿蒙生态一站式服务平台。

更多推荐