zdou0830 commited on
Commit
0ad0530
1 Parent(s): fc02c0e

Add application file

Browse files
Files changed (1) hide show
  1. Dockerfile +1 -99
Dockerfile CHANGED
@@ -1,102 +1,4 @@
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 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