change to cuda 12.1 for pytorch and tensorflow

isl-org · Sep 16, 2024 · 13699e8 · 13699e8
1 parent 7cdd7e3
commit 13699e8
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/docker/docker_build.sh b/docker/docker_build.sh
@@ -84,8 +84,8 @@ HOST_OPEN3D_ROOT="$(cd "$(dirname "${BASH_SOURCE[0]}")"/.. >/dev/null 2>&1 && pw
 CCACHE_VERSION=4.3
 CMAKE_VERSION=cmake-3.24.4-linux-x86_64
 CMAKE_VERSION_AARCH64=cmake-3.24.4-linux-aarch64
-CUDA_VERSION=12.3.2-cudnn9
-CUDA_VERSION_LATEST=12.3.2-cudnn9
+CUDA_VERSION=12.1.0-cudnn8
+CUDA_VERSION_LATEST=12.1.0-cudnn8
 
 print_usage_and_exit_docker_build() {
     echo "$__usage_docker_build"