Abhishek Thakur
commited on
Commit
·
e2ff59b
1
Parent(s):
e65f8ef
remove cuda toolkit
Browse files- Dockerfile +2 -2
Dockerfile
CHANGED
@@ -64,8 +64,8 @@ SHELL ["conda", "run","--no-capture-output", "-p","/app/env", "/bin/bash", "-c"]
|
|
64 |
|
65 |
RUN conda install pytorch torchvision torchaudio pytorch-cuda=12.1 -c pytorch -c nvidia && \
|
66 |
conda clean -ya && \
|
67 |
-
conda install -c "nvidia/label/cuda-12.1.0" cuda-nvcc && conda clean -ya
|
68 |
-
|
69 |
|
70 |
COPY --chown=1000:1000 . /app/
|
71 |
RUN make socket-kit.so
|
|
|
64 |
|
65 |
RUN conda install pytorch torchvision torchaudio pytorch-cuda=12.1 -c pytorch -c nvidia && \
|
66 |
conda clean -ya && \
|
67 |
+
conda install -c "nvidia/label/cuda-12.1.0" cuda-nvcc && conda clean -ya
|
68 |
+
# conda install -c "nvidia/label/cuda-12.1.0" cuda-toolkit && conda clean -ya
|
69 |
|
70 |
COPY --chown=1000:1000 . /app/
|
71 |
RUN make socket-kit.so
|