0
点赞
收藏
分享

微信扫一扫

IOS面试题编程机制 41-45

猎书客er 03-27 23:30 阅读 4

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)
举报

相关推荐

0 条评论