thistleknot
commented
1 year ago tried installing using .git just now, and same error
/home/user/env-10/lib/python3.10/site-packages/torch/functional.py:504: UserWarning: torch.meshgrid: in an upcoming release, it will be required to pass the indexing argument. (Triggered internally at ../aten/src/ATen/native/TensorShape.cpp:3483.)
return _VF.meshgrid(tensors, **kwargs) # type: ignore[attr-defined]
0%| | 0/631 [00:00<?, ?it/s][W NNPACK.cpp:64] Could not initialize NNPACK! Reason: Unsupported hardware.
0%| | 0/631 [05:40<?, ?it/s]
Traceback (most recent call last):
File "/home/user/env-10/bin/nougat", line 8, in <module>
sys.exit(main())
File "/home/user/env-10/lib/python3.10/site-packages/predict.py", line 130, in main
model_output = model.inference(image_tensors=sample)
File "/home/user/env-10/lib/python3.10/site-packages/nougat/model.py", line 577, in inference
last_hidden_state = self.encoder(image_tensors)
File "/home/user/env-10/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1501, in _call_impl
return forward_call(*args, **kwargs)
File "/home/user/env-10/lib/python3.10/site-packages/nougat/model.py", line 123, in forward
x = self.model.layers(x)
File "/home/user/env-10/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1501, in _call_impl
return forward_call(*args, **kwargs)
File "/home/user/env-10/lib/python3.10/site-packages/torch/nn/modules/container.py", line 217, in forward
input = module(input)
File "/home/user/env-10/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1501, in _call_impl
return forward_call(*args, **kwargs)
File "/home/user/env-10/lib/python3.10/site-packages/timm/models/swin_transformer.py", line 413, in forward
x = blk(x)
File "/home/user/env-10/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1501, in _call_impl
return forward_call(*args, **kwargs)
File "/home/user/env-10/lib/python3.10/site-packages/timm/models/swin_transformer.py", line 295, in forward
attn_windows = self.attn(x_windows, mask=self.attn_mask) # nW*B, window_size*window_size, C
File "/home/user/env-10/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1501, in _call_impl
return forward_call(*args, **kwargs)
File "/home/user/env-10/lib/python3.10/site-packages/timm/models/swin_transformer.py", line 183, in forward
attn = (q @ k.transpose(-2, -1))
RuntimeError: CUDA error: CUBLAS_STATUS_NOT_SUPPORTED when calling `cublasGemmStridedBatchedExFix(handle, opa, opb, (int)m, (int)n, (int)k, (void*)&falpha, a, CUDA_R_16BF, (int)lda, stridea, b, CUDA_R_16BF, (int)ldb, strideb, (void*)&fbeta, c, CUDA_R_16BF, (int)ldc, stridec, (int)num_batches, CUDA_R_32F, CUBLAS_GEMM_DEFAULT_TENSOR_OP)`
lukas-blecher
willocho
after pip install nougat-ocr
tried to run
nougat file.pdf -o .
I don't see a requirements... I'm on python 3.10, using ubuntu 22 in wsl. I am able to run nvidia-smi