Failed building wheel for pytorch-fast-transformers

[jinfeng-data]

Hi, I am trying to install the MolFormer environment. When I run "pip install transformers==4.6.0 pytorch-lightning==1.1.5 pytorch-fast-transformers==0.4.0 datasets==1.6.2 jupyterlab==3.4.0 ipywidgets==7.7.0 bertviz==1.4.0", it fails, and the error messages are like the following:

_Building wheels for collected packages: pytorch-fast-transformers
Building wheel for pytorch-fast-transformers (setup.py) ... error
error: subprocess-exited-with-error

× python setup.py bdist_wheel did not run successfully.
│ exit code: 1
╰─> [151 lines of output]
No CUDA runtime is found, using CUDA_HOME='/usr/local/cuda-11.1'
running bdist_wheel
running build
running build_py
..........
/home/jinfeng/soft/miniconda3/envs/MolTran_CUDA11/compiler_compat/ld: cannot find -lc10_cuda: 没有那个文件或目录
/home/jinfeng/soft/miniconda3/envs/MolTran_CUDA11/compiler_compat/ld: cannot find -ltorch_cuda: 没有那个文件或目录
collect2: 错误：ld 返回 1
error: command 'g++' failed with exit status 1
[end of output]

note: This error originates from a subprocess, and is likely not a problem with pip.
ERROR: Failed building wheel for pytorch-fast-transformers
Running setup.py clean for pytorch-fast-transformers
Failed to build pytorch-fast-transformers
ERROR: Could not build wheels for pytorch-fast-transformers, which is required to install pyproject.toml-based projects

What should I do to fix the above problem ? Thanks in advance.

[Basspoom]

I encounter a similar error:
"pip install -i https://pypi.tuna.tsinghua.edu.cn/simple pytorch-fast-transformers==0.4.0
Looking in indexes: https://pypi.tuna.tsinghua.edu.cn/simple
Collecting pytorch-fast-transformers==0.4.0
Using cached https://pypi.tuna.tsinghua.edu.cn/packages/ea/bc/00f597fefeab6341114c41045c1b232c38436738e0e8eac1bc9d5e9d5962/pytorch-fast-transformers-0.4.0.tar.gz (93 kB)
Preparing metadata (setup.py) ... done
......
Building wheels for collected packages: pytorch-fast-transformers
Building wheel for pytorch-fast-transformers (setup.py) ... error
error: subprocess-exited-with-error

× python setup.py bdist_wheel did not run successfully.
│ exit code: 1
╰─> [148 lines of output]
......
[end of output]

note: This error originates from a subprocess, and is likely not a problem with pip.
ERROR: Failed building wheel for pytorch-fast-transformers
Running setup.py clean for pytorch-fast-transformers
Failed to build pytorch-fast-transformers
ERROR: ERROR: Failed to build installable wheels for some pyproject.toml based projects (pytorch-fast-transformers)"

I fix this issue by run command "pip install torch torchvision torchaudio".
Then try "pip install pytorch-fast-transformers==0.4.0" again.