search

FangyunWei / SLRT

236 stars 46 forks source link

TwoStreamNetwork environment has conflict #20

Closed VinnyCSY closed 1 year ago

VinnyCSY commented 1 year ago

I am using docker, and there is a problem in version of sentencepiece and tensorflow. There needs sentencepiece to use bert. Is there solution for this or any dockerfile to make this environment? Thank you.

VinnyCSY commented 1 year ago

This is a code of my dockerfile. I use sentencepiece 0.1.90 and tensorflow 2.2.0

FROM pytorch/pytorch:1.9.0-cuda11.1-cudnn8-devel ARG DEBIAN_FRONTEND=noninteractive RUN apt-key del 7fa2af80 && apt-key adv --keyserver keyserver.ubuntu.com --recv-keys F60F4B3D7FA2AF80 RUN apt-key adv --keyserver keyserver.ubuntu.com --recv-keys A4B469963BF863CC

Install apt dependencies as root

RUN apt-get update && apt-get install -y \ git \ wget \ ffmpeg \ libsm6 \ libxext6 \ nano \ sudo

Install python dependencies

RUN pip install timm==0.4.12 matplotlib pandas \ torchvision==0.10.0 \ numpy==1.19.2 \ scipy \ scikit-learn==0.22.2.post1 \ opencv-python==4.4.0.46 \ scikit-image \ tensorboard \ tensorflow==2.2.0 \ protobuf==3.19.4 \ wandb \ transformers==4.11.3 \ portalocker \ sentencepiece==0.1.90

VinnyCSY commented 1 year ago

Resolved!