diff --git a/docker/Dockerfile.x64 b/docker/Dockerfile.x64
index 05cad5c7..26bb85f4 100644
--- a/docker/Dockerfile.x64
+++ b/docker/Dockerfile.x64
@@ -16,9 +16,9 @@ RUN apt-get update && apt-get install -y \
     git
 
 # Install PyTorch
-RUN pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118
 COPY requirements.txt /tmp/requirements.txt
 RUN pip3 install -r /tmp/requirements.txt
+RUN pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118
 
 # Install other Python packages
 RUN pip3 install cupy-cuda11x scikit-learn
diff --git a/requirements.txt b/requirements.txt
index 76a0b9d2..fa3f3767 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -7,6 +7,7 @@ simple-parsing
 scikit-image==0.19
 matplotlib
 catkin-tools
+networkx==3.0
 # cupy
 
 ###### Requirements with Version Specifiers ######`