在运行pytorch程序时，出现GPU内存溢出，使用nvidia-smi查看显卡只用了一个：
因此，可以切换到其他显卡上:
在程序最上方添加：
import osos.environ["CUDA_VISIBLE_DEVICES"] = '1'import torchfrom torch import optim, nnimport visdomimport torchvisionfrom torch.utils.data import DataLoaderfrom pokemon import Pokemon# from resnet import ResNet18from torchvision.models import resnet18from utils import Flatten
注意要在import torch之前添加，不然不生效。
报错：RuntimeError: CUDA out of memory. Tried to allocate 5.66 GiB (GPU 0; 12.00 GiB total
网上查询了很多案例都是重复的，并不能处理我的问题。最后看到了[https://www.javaear.com/question/41120920.html][https_www
方法一： 仅需减小batchsize 改文件的配置cfg的batchsize=4，一般在cfg文件下的查找batch或batchsize，将batchsize调小后，再
1， 版本 2，问题 remote: fatal: Out of memory, malloc failed (tried to allocate 2604
[参考网址][Link 1] 其实 \loss.backward() [Link 1]: https://blog.csdn.net/qq_36213248/articl
在运行pytorch程序时，出现GPU内存溢出，使用`nvidia-smi`查看显卡只用了一个： ![在这里插入图片描述][watermark_type_ZmFuZ3poZ
写在前面（ 已解决） 记录一下很诡异的情况，实验室两台同样的电脑做服务器，相同的代码在一直使用的机子上是 ok 的，移到另一台配置好的电脑就开始在 `model=mode
笔者近日利用pytorch框架对图像采用Resnet50模型提取特征的时候，利用到了GPU，发现报错如下： RuntimeError: CUDA out of memory.
最近写代码，指定显卡的时候出现了`CUDA error：out of memory`的错误，用`nvidia-smi`命令查看发现该显卡并没有被占用呀，查了资料后才知道是调用的
错误 Traceback (most recent call last): File "train.py", line 137, in <module
ゞ 浴缸里的玫瑰
水深无声
小灰灰
我会带着你远行
「爱情、让人受尽委屈。」
怼烎@
还没有评论，来说两句吧...