Install requirements before PyTorch

leggedrobotics · Apr 24, 2024 · dd3d9a2 · dd3d9a2
1 parent 8de87ce
commit dd3d9a2
Showing 1 changed file with 1 addition 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 networkx==3.0 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