From 514a09b8f5efeb5964cad777886011cfff6c2d19 Mon Sep 17 00:00:00 2001 From: "ben.sturgis" Date: Wed, 7 Aug 2024 09:31:52 +0200 Subject: [PATCH] Specify kornia version 0.6.5 in cu116.Dockerfile --- docker/cu116.Dockerfile | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/docker/cu116.Dockerfile b/docker/cu116.Dockerfile index f68e4bdf7..b628a8fdd 100644 --- a/docker/cu116.Dockerfile +++ b/docker/cu116.Dockerfile @@ -62,7 +62,7 @@ RUN pip3 install torch==1.13.1+cu116 torchvision==0.14.1+cu116 torchaudio==0.13. ENV TORCH_CUDA_ARCH_LIST="3.5;5.0;6.0;6.1;7.0;7.5;8.0;8.6+PTX" # OpenPCDet -RUN pip3 install numpy==1.23.0 llvmlite numba tensorboardX easydict pyyaml scikit-image tqdm SharedArray open3d mayavi av2 kornia pyquaternion +RUN pip3 install numpy==1.23.0 llvmlite numba tensorboardX easydict pyyaml scikit-image tqdm SharedArray open3d mayavi av2 kornia==0.6.5 pyquaternion RUN pip3 install spconv-cu116 RUN git clone https://github.com/open-mmlab/OpenPCDet.git