pigbreeder / CodeMemo

1 stars 0 forks source link

pytorch 坑 #2

Open testpppppp opened 1 year ago

testpppppp commented 1 year ago

在jupyter中调试,dataloader 的num_worker=0 不然 unpickling error on dataloading

使用argparse,args, unknown = parser.parse_known_args()不然 ipykernel_launcher.py: error: unrecognized arguments

快速判断能不能用,不行就换cuda版本 import torch device = torch.device("cuda") torch.rand(10).to(device)

大数据逐行读取 https://blog.csdn.net/u011550545/article/details/87842760