diff options
Diffstat (limited to '')
| -rw-r--r-- | Dockerfile | 15 | ||||
| -rw-r--r-- | docker-compose.yml | 1 |
2 files changed, 13 insertions, 3 deletions
@@ -3,11 +3,20 @@ RUN apk add openssl WORKDIR /certs RUN openssl req -nodes -new -x509 -subj="/C=US/ST=Denial/L=springfield/O=Dis/CN=localhost" -keyout server.key -out server.cert && cat server.key server.cert > cert.pem -FROM python:3.10.5-slim-buster -RUN pip install jupyterlab scipy numpy poetry numba torch tensorflow transformers huggingface-hub +FROM nvidia/cuda:11.7.0-runtime-ubuntu20.04 +ENV DEBIAN_FRONTEND=noninteractive +RUN apt-get update &&\ + apt-get install -y software-properties-common &&\ + add-apt-repository ppa:deadsnakes/ppa &&\ + apt-get update &&\ + apt-get install -y python3.10 python3-pip +RUN pip install pysocks +ENV HTTP_PROXY=socks5://172.17.0.1:9995 +ENV HTTPS_PROXY=socks5://172.17.0.1:9995 +RUN pip install jupyterlab scipy numpy poetry numba torch tensorflow transformers huggingface-hub ipywidgets pandas RUN apt-get update && apt-get install -y curl wget unzip RUN curl -fsSL https://deb.nodesource.com/setup_16.x | bash - RUN apt-get -y install nodejs -RUN pip install ipywidgets pandas pysocks +# RUN pip install ipywidgets pandas pysocks # RUN jupyter labextension install jupyterlab_vim COPY --from=certbuilder /certs /certs diff --git a/docker-compose.yml b/docker-compose.yml index 1acf735..79df23f 100644 --- a/docker-compose.yml +++ b/docker-compose.yml @@ -21,6 +21,7 @@ services: - HTTP_PROXY=socks5://172.17.0.1:9995 - HTTPS_PROXY=socks5://172.17.0.1:9995 - NO_PROXY="localhost, 127.0.0.1" + - NVIDIA_VISIBLE_DEVICES=all runtime: nvidia volumes: devijlab-vault: |