diff --git a/DockerfileServer b/DockerfileServer index a6fdaa6..ee8f480 100644 --- a/DockerfileServer +++ b/DockerfileServer @@ -14,7 +14,7 @@ ENV PATH=$PATH:/usr/local/go/bin ENV GOPATH=/go RUN bash -c 'curl -L "https://storage.googleapis.com/tensorflow/libtensorflow/libtensorflow-cpu-linux-x86_64-2.9.1.tar.gz" | tar -C /usr/local -xz' -# RUN bash -c 'curl -L "https://storage.googleapis.com/tensorflow/libtensorflow/libtensorflow-cpu-linux-x86_64-2.13.1.tar.gz" | tar -C /usr/local -xz' +RUN bash -c 'curl -L "https://storage.googleapis.com/tensorflow/libtensorflow/libtensorflow-cpu-linux-x86_64-2.15.0.tar.gz" | tar -C /usr/local -xz' RUN ldconfig RUN ln -s /usr/bin/python3 /usr/bin/python diff --git a/requirements.txt b/requirements.txt index e0db44b..a4c8eec 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,5 +1,4 @@ -# tensorflow[and-cuda] == 2.15.1 -tensorflow[and-cuda] == 2.9.1 +tensorflow[and-cuda] == 2.15.1 pandas # Make sure to install the nvidia pyindex first # nvidia-pyindex diff --git a/run.sh b/run.sh index c887593..f216200 100755 --- a/run.sh +++ b/run.sh @@ -1,2 +1,2 @@ #!/bin/bash -podman run --rm --network host --gpus all -ti -v $(pwd):/app -e "TERM=xterm-256color" fyp-server bash +podman run --rm --network host --gpus all --name fyp-server -it -v $(pwd):/app -e "TERM=xterm-256color" fyp-server bash