From c94b90a5074aaed25c7d21e8e331bcbe2982dd57 Mon Sep 17 00:00:00 2001 From: terminaldweller Date: Sun, 10 Jul 2022 13:12:57 +0430 Subject: updated --- Dockerfile | 15 ++++++++++++--- docker-compose.yml | 1 + 2 files changed, 13 insertions(+), 3 deletions(-) diff --git a/Dockerfile b/Dockerfile index a81d555..77bbc3f 100644 --- a/Dockerfile +++ b/Dockerfile @@ -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: -- cgit v1.2.3