1.查看显卡:
第一步:同时按下键盘的win+r键,输入cmd
第二步:随后输入
nvidia-smi
2.查看下载安装的包:
conda list
conda list
3.问题报错
PyTorch出现:RuntimeError: An attempt has been made to start a new process...报错
出错原因:网上查找结果,原因是多进程的原因
解决方法:
1. 代码在运行 epoch
之前,加上 if __name__=='__main__'
2, 不使用多线程,即去掉 num_workers 参数,或设置 num_workers=0。
dataset = torch.utils.data.DataLoader(trainset, batch_size=opt.batch_size,
shuffle=True, num_workers=0, drop_last=True)