RuntimeError: 0 INTERNAL ASSERT FAILED at "../torch/csrc/jit/ir/alias_analysis.cpp":615, please report a bug to PyTorch. We don't have an op for vulkan_prepack::create_linear_context but it isn't a special case. Argument types: Tensor, Tensor, about pytorch HOT 2 CLOSED

I just noticed something, which seems not right to me, as I mentioned in the Version Detailed:

Torch Version
[conda] torch 2.4.0a0+git57baae9 pypi_0 pypi
[pip3] torch==2.0.1+cu118

And when I see under Interactive Python session:
torch version is 2.0.1+cu118

(pytorch_vulkan) n.bansal1@SRA0401:~/projects/utils$ python
Python 3.10.14 (main, May  6 2024, 19:42:50) [GCC 11.2.0] on linux
Type "help", "copyright", "credits" or "license" for more information.
>>> import torch
>>> torch.__version__
'2.0.1+cu118'

from pytorch.

Removing the pip installed torch, removes the error, I am receiving.
Closing this issue for now.

from pytorch.