diff --git a/Dockerfile-GPU b/Dockerfile-GPU index 91b0d04..2e847d1 100644 --- a/Dockerfile-GPU +++ b/Dockerfile-GPU @@ -16,6 +16,4 @@ RUN pip install pyg_lib torch_scatter torch_sparse torch_cluster torch_spline_co RUN pip install absl-py autopep8 numpy==1.23.1 dm-tree matplotlib pyevtk pytest tqdm toml RUN pip install -r requirements.txt - - CMD ["/bin/bash"] \ No newline at end of file