No CUDA GPUs are available

2023-12-13 17:27:00


前言

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

文章来源:https://blog.csdn.net/weixin_42504788/article/details/134895512
本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。