Merge pull request #94 from yyagin/main

Networkx installation error fix
leggedrobotics · Apr 24, 2024 · 9617d84 · 9617d84
2 parents ee9a155 + dd3d9a2
commit 9617d84
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git 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
@@ -7,6 +7,7 @@ simple-parsing
 scikit-image==0.19
 matplotlib
 catkin-tools
+networkx==3.0
 # cupy
 
 ###### Requirements with Version Specifiers ######`