Abhishek Thakur commited on
Commit
e2ff59b
·
1 Parent(s): e65f8ef

remove cuda toolkit

Browse files
Files changed (1) hide show
  1. 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
- 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
 
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