Skip to content

Commit

Permalink
Added keras
Browse files Browse the repository at this point in the history
  • Loading branch information
@krishnaa423
krishnaa423 committed Jul 12, 2024
1 parent 3a1b47d commit 34fa0f7
Show file tree
Hide file tree
Showing 2 changed files with 20 additions and 0 deletions.
10 changes: 10 additions & 0 deletions Dockerfile_cuda
View file
Original file line number Diff line number Diff line change
Expand Up @@ -80,6 +80,12 @@ RUN source ~/.bashrc \
RUN source ~/.bashrc \
&& mamba install -y pytorch torchvision torchaudio pytorch-cuda=12.1 -c pytorch -c nvidia

RUN source ~/.bashrc \
&& mamba install -y tensorboard

RUN source ~/.bashrc \
&& pip3 install -q keras

RUN source ~/.bashrc \
&& pip3 install -q torch_geometric pyg_lib torch_scatter torch_sparse torch_cluster torch_spline_conv \
-f https://data.pyg.org/whl/torch-2.3.0+cu121.html
Expand All @@ -93,6 +99,10 @@ RUN source ~/.bashrc \
RUN source ~/.bashrc \
&& pip3 install -q mp_api

# Package development.
RUN source ~/.bashrc \
&& pip3 install -q twine

# Finally set the working directory to home directory.
WORKDIR /app

Expand Down
10 changes: 10 additions & 0 deletions Dockerfile_mpich
View file
Original file line number Diff line number Diff line change
Expand Up @@ -76,6 +76,12 @@ RUN source ~/.bashrc \
RUN source ~/.bashrc \
&& mamba install -y pytorch torchvision torchaudio cpuonly -c pytorch

RUN source ~/.bashrc \
&& mamba install -y tensorboard

RUN source ~/.bashrc \
&& pip3 install -q keras

RUN source ~/.bashrc \
&& pip3 install -q torch_geometric

Expand All @@ -88,6 +94,10 @@ RUN source ~/.bashrc \
RUN source ~/.bashrc \
&& pip3 install -q mp_api

# Package development.
RUN source ~/.bashrc \
&& pip3 install -q twine

# Finally set the working directory to home directory.
WORKDIR /app

Expand Down

0 comments on commit 34fa0f7

Please sign in to comment.