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