You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
On ImageNet set, it occured as following:
File "train.py", line 245, in
main()
File "train.py", line 241, in main
engine.train(h, train_loader, opt.epochs, optimizer)
File "build/bdist.linux-x86_64/egg/torchnet/engine/engine.py", line 39, in train
File "/usr/local/lib/python2.7/dist-packages/torch/optim/sgd.py", line 72, in step
loss = closure()
File "build/bdist.linux-x86_64/egg/torchnet/engine/engine.py", line 28, in closure
File "train.py", line 177, in h
y = data_parallel(f, inputs, params, stats, sample[2], np.arange(opt.ngpu))
File "/home/yq/work/face_class/diracnets/diracnet.py", line 51, in data_parallel
return f(input, params, stats, mode)
File "/home/yq/work/face_class/diracnets/diracnet.py", line 182, in f
o = F.linear(o.view(o.size(0), -1), params['fc.weight'], params['fc.bias'])
File "/usr/local/lib/python2.7/dist-packages/torch/nn/functional.py", line 449, in linear
return state(input, weight) if bias is None else state(input, weight, bias)
File "/usr/local/lib/python2.7/dist-packages/torch/nn/functions/linear.py", line 10, in forward
output.addmm(0, 1, input, weight.t())
RuntimeError: cublas runtime error : library not initialized at /b/wheel/pytorch-src/torch/lib/THC/THCGeneral.c:394
The text was updated successfully, but these errors were encountered:
On ImageNet set, it occured as following:
File "train.py", line 245, in
main()
File "train.py", line 241, in main
engine.train(h, train_loader, opt.epochs, optimizer)
File "build/bdist.linux-x86_64/egg/torchnet/engine/engine.py", line 39, in train
File "/usr/local/lib/python2.7/dist-packages/torch/optim/sgd.py", line 72, in step
loss = closure()
File "build/bdist.linux-x86_64/egg/torchnet/engine/engine.py", line 28, in closure
File "train.py", line 177, in h
y = data_parallel(f, inputs, params, stats, sample[2], np.arange(opt.ngpu))
File "/home/yq/work/face_class/diracnets/diracnet.py", line 51, in data_parallel
return f(input, params, stats, mode)
File "/home/yq/work/face_class/diracnets/diracnet.py", line 182, in f
o = F.linear(o.view(o.size(0), -1), params['fc.weight'], params['fc.bias'])
File "/usr/local/lib/python2.7/dist-packages/torch/nn/functional.py", line 449, in linear
return state(input, weight) if bias is None else state(input, weight, bias)
File "/usr/local/lib/python2.7/dist-packages/torch/nn/functions/linear.py", line 10, in forward
output.addmm(0, 1, input, weight.t())
RuntimeError: cublas runtime error : library not initialized at /b/wheel/pytorch-src/torch/lib/THC/THCGeneral.c:394
The text was updated successfully, but these errors were encountered: