Abhishek Thakur
commited on
Commit
·
9987f71
1
Parent(s):
ad43a34
add cuda toolkit
Browse files- Dockerfile +2 -1
Dockerfile
CHANGED
|
@@ -64,7 +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 |
COPY --chown=1000:1000 . /app/
|
| 70 |
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
|