RuntimeError: CUDA out of memory. Tried to allocate 98.00 MiB (GPU 0； 31.75 GiB total capacity；

在运行pytorch程序时，出现GPU内存溢出，使用nvidia-smi查看显卡只用了一个：

因此，可以切换到其他显卡上:
在程序最上方添加：

import os
os.environ["CUDA_VISIBLE_DEVICES"] = '1'

import torch
from torch import optim, nn
import visdom
import torchvision
from torch.utils.data import DataLoader

from pokemon import Pokemon
# from    resnet import ResNet18
from torchvision.models import resnet18

from utils import Flatten

注意要在import torch之前添加，不然不生效。