File size: 265 Bytes
eec42bd |
1 2 3 4 5 6 7 |
FROM tensorflow/tensorflow:2.4.1-gpu-jupyter
RUN apt update && apt install -y nvidia-modprobe
RUN pip install adabelief-tf pandas imageio imageio-ffmpeg seaborn sklearn tensorflow_addons tqdm
RUN pip install keras_applications tensorflowjs
RUN pip install tf2onnx
|