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
本文来自互联网用户投稿，该文观点仅代表作者本人，不代表本站立场。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。