Skip to content
Snippets Groups Projects
Select Git revision
  • master
  • skyflynil
  • sdtblck
  • selfattention
  • fp16
  • swarm
  • tpu
7 results

Dockerfile

Blame
  • Dockerfile 688 B
    # Copyright (c) 2020, NVIDIA CORPORATION.  All rights reserved.
    #
    # NVIDIA CORPORATION and its licensors retain all intellectual property
    # and proprietary rights in and to this software, related documentation
    # and any modifications thereto.  Any use, reproduction, disclosure or
    # distribution of this software and related documentation without an express
    # license agreement from NVIDIA CORPORATION is strictly prohibited.
    
    FROM tensorflow/tensorflow:1.14.0-gpu-py3
    
    RUN pip install scipy==1.3.3
    RUN pip install requests==2.22.0
    RUN pip install Pillow==6.2.1
    RUN pip install h5py==2.9.0
    RUN pip install imageio==2.9.0
    RUN pip install imageio-ffmpeg==0.4.2
    RUN pip install tqdm==4.49.0