Spaces:
Sleeping
Sleeping
File size: 1,917 Bytes
6ece75e |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 |
ENV PATH=/opt/conda/bin:/opt/cmake-3.14.6-Linux-x86_64/bin/:/usr/local/mpi/bin:/usr/local/nvidia/bin:/usr/local/cuda/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin:/usr/local/ucx/bin:/opt/tensorrt/bin
ENV CUDA_VERSION=11.3.1.005
ENV CUDA_DRIVER_VERSION=465.19.01
ENV CUDA_CACHE_DISABLE=1
ENV _CUDA_COMPAT_PATH=/usr/local/cuda/compat
ENV ENV=/etc/shinit_v2
ENV BASH_ENV=/etc/bash.bashrc
ENV NVIDIA_REQUIRE_CUDA=cuda>=9.0
ENV NCCL_VERSION=2.9.9
ENV CUBLAS_VERSION=11.5.1.109
ENV CUFFT_VERSION=10.4.2.109
ENV CURAND_VERSION=10.2.4.109
ENV CUSPARSE_VERSION=11.6.0.109
ENV CUSOLVER_VERSION=11.1.2.109
ENV NPP_VERSION=11.3.3.95
ENV NVJPEG_VERSION=11.5.0.109
ENV CUDNN_VERSION=8.2.1.32
ENV TRT_VERSION=7.2.3.4+cuda11.1.0.024
ENV TRTOSS_VERSION=21.06
ENV NSIGHT_SYSTEMS_VERSION=2021.2.1.58
ENV NSIGHT_COMPUTE_VERSION=2021.1.1.5
ENV DALI_VERSION=1.2.0
ENV DALI_BUILD=2356513
ENV DLPROF_VERSION=21.06
ENV LD_LIBRARY_PATH=/opt/hpcx/nccl_rdma_sharp_plugin/lib:/usr/local/cuda/compat/lib:/usr/local/nvidia/lib:/usr/local/nvidia/lib64
ENV NVIDIA_VISIBLE_DEVICES=all
ENV NVIDIA_DRIVER_CAPABILITIES=compute,utility,video
ENV GDRCOPY_VERSION=2.2
ENV HPCX_VERSION=2.8.2rc3
ENV MOFED_VERSION=5.2-rdmacore32.1
ENV OPENUCX_VERSION=1.10.1
ENV OPENMPI_VERSION=4.1.1rc1
ENV RDMACORE_VERSION=32.1
ENV LIBRARY_PATH=/usr/local/cuda/lib64/stubs:
ENV PYTORCH_BUILD_VERSION=1.9.0a0+c3d40fd
ENV PYTORCH_VERSION=1.9.0a0+c3d40fd
ENV PYTORCH_BUILD_NUMBER=0
ENV NVIDIA_PYTORCH_VERSION=21.06
ENV NVM_DIR=/usr/local/nvm
ENV JUPYTER_PORT=8888
ENV TENSORBOARD_PORT=6006
ENV COCOAPI_VERSION=2.0+nv0.5.1
ENV PYTHONIOENCODING=utf-8
ENV LC_ALL=C.UTF-8
ENV NVIDIA_BUILD_ID=24129275
ENV NCCL_ASYNC_ERROR_HANDLING=1
EXPOSE 6006/tcp
EXPOSE 8888/tcp
WORKDIR /workspace
ENTRYPOINT ["/usr/local/bin/nvidia_entrypoint.sh"]
COPY . .
CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "7860"]
|