FROM continuumio/miniconda3


RUN mkdir /root/docker_dir/search_capability -p
WORKDIR /root/docker_dir/search_capability

COPY . .
RUN chmod 777 scripts/data/*/*.sh
RUN chmod 777 scripts/model/*/*.sh

RUN conda create -n search_capability python=3.11
RUN conda install conda-forge::transformers
RUN conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch -c nvidia
RUN conda install -c conda-forge wandb tqdm openai lightning sentencepiece tenacity accelerate
RUN pip install -U bitsandbytes


#RUN conda install pytorch==2.2.0 torchvision==0.17.0 torchaudio==2.2.0 pytorch-cuda=11.8 -c pytorch -c nvidia
#RUN echo 'export BNB_CUDA_VERSION=118' >> ~/.bashrc
#RUN echo 'export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/opt/conda/lib/' >> ~/.bashrc
#RUN conda install pytorch==2.1.1 torchvision==0.16.1 torchaudio==2.1.1 pytorch-cuda=11.8 -c pytorch -c nvidia
#RUN conda create -n search_capability python=3.9
