RUN wget --no-check-certificate--content-disposition'https://cancon.hpccube.com:65024/directlink/4/flash_attn/DAS1.2/flash_attn-2.0.4+das.opt2.dtk24042-cp310-cp310-manylinux_2_28_x86_64.whl'
RUN pip install flash_attn-*.whl &&rm flash_attn-*.whl
#########
# Prod #
#########
FROM jupyterlab-pytorch:2.3.0-ubuntu22.04-dtk24.04.2-py3.10-devel