クッキーの日記

数式が崩れていたらその記事単体を表示すると解消するかもしれないです。

雑記

以下の Pipfle と Pipfile.lock から pipenv を install すると fast_transformer はインストールできた。ただし fast_transformer 以外のパッケージもごちゃごちゃ入っている。

$ pipenv install
$ pipenv run python
>>> import fast_transformers.causal_product.causal_product_cuda

あと上記のリポジトリには pytest にコードのはぎれを置いてあるが通しで実行すると tensorflow 系のテストが通らない。個々のテストファイルだけを実行すると通る。pytest ではまれによくある。

 tensorflow.python.framework.errors_impl.FailedPreconditionError: Failed to allocate scratch buffer for device 0