Spaces:
Sleeping
Sleeping
Add application file
Browse files- Dockerfile +1 -99
Dockerfile
CHANGED
@@ -1,102 +1,4 @@
|
|
1 |
-
|
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 COCOAPI_VERSION=2.0+nv0.5.1
|
84 |
-
|
85 |
-
ENV PYTHONIOENCODING=utf-8
|
86 |
-
|
87 |
-
ENV LC_ALL=C.UTF-8
|
88 |
-
|
89 |
-
ENV NVIDIA_BUILD_ID=24129275
|
90 |
-
|
91 |
-
ENV NCCL_ASYNC_ERROR_HANDLING=1
|
92 |
-
|
93 |
-
EXPOSE 6006/tcp
|
94 |
-
|
95 |
-
EXPOSE 8888/tcp
|
96 |
-
|
97 |
-
WORKDIR /workspace
|
98 |
-
ENTRYPOINT ["/usr/local/bin/nvidia_entrypoint.sh"]
|
99 |
-
|
100 |
|
101 |
COPY . .
|
102 |
|
|
|
1 |
+
FROM pengchuanzhang/pytorch:ubuntu20.04_torch1.9-cuda11.3-nccl2.9.9:latest
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
2 |
|
3 |
COPY . .
|
4 |
|