Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

cublas runtime error #7

Open
eeric opened this issue Jun 17, 2017 · 2 comments
Open

cublas runtime error #7

eeric opened this issue Jun 17, 2017 · 2 comments

Comments

@eeric
Copy link

eeric commented Jun 17, 2017

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

@zhenheny
Copy link

try sudo rm -rf ~/.nv

@eeric
Copy link
Author

eeric commented Jun 16, 2018

@zhenheny, thanks, so I didn't pay attention to it at present.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants