From e22352a592c7ace8fb92a6223d4624532d4c0991 Mon Sep 17 00:00:00 2001
From: Tyler <admin@meow.tf>
Date: Tue, 6 Dec 2022 07:36:15 -0500
Subject: [PATCH] Install tensorflow-gpu

---
 Dockerfile.gpu | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/Dockerfile.gpu b/Dockerfile.gpu
index 75325e0..13d763d 100644
--- a/Dockerfile.gpu
+++ b/Dockerfile.gpu
@@ -12,7 +12,7 @@ FROM python-ubuntu
 
 RUN apt-get update &&  apt-get install liblapack-dev libpng-dev libfreetype6-dev libqhull-dev git pkg-config portaudio19-dev swig libpulse-ocaml-dev gfortran libopenblas-dev libatlas-base-dev -y
 
-RUN mkdir /training && pip3 install precise-lite
+RUN mkdir /training && pip3 install precise-lite && pip3 install tensorflow-gpu==2.3.1
 
 WORKDIR /training