diff --git a/docker/Dockerfile.x64 b/docker/Dockerfile.x64 index 26bb85f4..41165273 100644 --- a/docker/Dockerfile.x64 +++ b/docker/Dockerfile.x64 @@ -18,7 +18,8 @@ RUN apt-get update && apt-get install -y \ # Install PyTorch 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 +RUN pip install torch==1.13.1+cu117 torchvision==0.14.1+cu117 torchaudio==0.13.1 --index-url https://download.pytorch.org/whl/cu117 + # Install other Python packages RUN pip3 install cupy-cuda11x scikit-learn @@ -43,4 +44,4 @@ RUN apt install -y ros-noetic-pybind11-catkin \ ros-noetic-ros-numpy RUN pip3 install catkin-tools -ENV TURTLEBOT3_MODEL=waffle \ No newline at end of file +ENV TURTLEBOT3_MODEL=waffle