aboutsummaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
authorterminaldweller <thabogre@gmail.com>2022-07-10 08:42:57 +0000
committerterminaldweller <thabogre@gmail.com>2022-07-10 08:42:57 +0000
commitc94b90a5074aaed25c7d21e8e331bcbe2982dd57 (patch)
treecac02be35efa6377b137f0df4469433b435e5f26
parentnvidia run time (diff)
downloadjupyter-notebook-docker-compose-c94b90a5074aaed25c7d21e8e331bcbe2982dd57.tar.gz
jupyter-notebook-docker-compose-c94b90a5074aaed25c7d21e8e331bcbe2982dd57.zip
updated
-rw-r--r--Dockerfile15
-rw-r--r--docker-compose.yml1
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: