解决RuntimeError: CUDA error: out of memory

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

$ nvidia-smi -l

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

CUDA_VISIBLE_DEVICES=2 python my_script.py

可以正常运行

相关资源

相似解决方案

热门标签

Java Python linux javascript Mysql C# Docker 算法前端 SpringBoot Redis Vue spring 设计模式 .net core .net kubernetes c++ 数据库数据结构大数据 js 机器学习微服务 Android Go 程序员面试 JVM ASP.net core 云原生人工智能后端 PHP git CSS golang k8s Nginx Django mybatis 深度学习多线程 React 架构 devops 爬虫云计算 Spring Boot LeetCode