mirror of
https://github.com/kodxana/madiator-docker-runpod.git
synced 2024-11-28 21:30:12 +01:00
21 lines
473 B
Docker
21 lines
473 B
Docker
FROM runpod/pytorch:2.0.1-py3.10-cuda11.8.0-devel
|
|
|
|
ENV NCCL_DEBUG=TRACE
|
|
|
|
WORKDIR /verify-nccl
|
|
|
|
# Install additional packages
|
|
RUN apt-get update && apt-get install -y --no-install-recommends \
|
|
nano \
|
|
pciutils \
|
|
&& rm -rf /var/lib/apt/lists/*
|
|
|
|
RUN git clone https://github.com/NVIDIA/cuda-samples.git && \
|
|
cd cuda-samples/Samples/0_Introduction/simpleP2P && \
|
|
make
|
|
|
|
COPY check_nccl.sh .
|
|
RUN chmod +x check_nccl.sh
|
|
|
|
# Start Container
|
|
CMD tail -f /dev/null
|