diff --git a/icecube_ml/Dockerfile_ubuntu_22.04_cuda11.8_cudann8_icetray_graphnet-devel b/icecube_ml/Dockerfile_ubuntu_22.04_cuda11.8_cudann8_icetray_graphnet-devel index 397009f..337c7b3 100644 --- a/icecube_ml/Dockerfile_ubuntu_22.04_cuda11.8_cudann8_icetray_graphnet-devel +++ b/icecube_ml/Dockerfile_ubuntu_22.04_cuda11.8_cudann8_icetray_graphnet-devel @@ -4,6 +4,7 @@ FROM icecube/icetray-ml:icetray-${VERSION}-cuda11.8-cudnn8-ubuntu22.04 RUN pip3 install --upgrade pip RUN pip3 install --upgrade astropy RUN pip3 install --ignore-installed PyYAML +RUN pip3 install --upgrade pandas RUN pip3 install --find-links https://download.pytorch.org/whl/torch_stable.html torch==2.0.1+cu118 --find-links https://data.pyg.org/whl/torch-2.0.0+cu118.html