2022-05-01から1日間の記事一覧

雑記: [PyTorch on WSL2] fast_transformers.causal_product.causal_product_cuda が import できなかったのが解消した話

手元の環境(CUDA 11.6 + cuDNN v8.2.0 on Ubuntu20.04 on WSL2)では pytorch-fast-transformers を pip でインストールしたところ一部の機能は使用できるようだが fast_transformers.causal_product.causal_product_cuda が import できなかった。手元では…