明明跑了一段时间?跑过一次完整的?怎么就出现这个报错呢?代码也未改动?而这就是现实!
- import os
- os.environ['CUDA_VISIBLE_DEVICES'] = "2"
- import torch
conda install pytorch torchvision cudatoolkit=10.1
- # True:每次返回的卷积算法将是确定的,即默认算法。
- torch.backends.cudnn.deterministic = True
- # 程序在开始时花额外时间,为整个网络的每个卷积层搜索最适合它的卷积实现算法
- # 实现网络的加速。
- torch.backends.cudnn.benchmark = True