Updated Jupyter Notebooks

Tim-Salzmann · Mar 24, 2024 · 32e5439 · 32e5439
1 parent cc94109
commit 32e5439
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 7 deletions.
diff --git a/examples/fish_turbulent_flow/Fish_Turbulent_Flow.ipynb b/examples/fish_turbulent_flow/Fish_Turbulent_Flow.ipynb
@@ -145,7 +145,8 @@
       ],
       "source": [
         "# @title\n",
-        "!pip install torch scikit-build cmake ninja\n",
+        "!pip install torch --index-url https://download.pytorch.org/whl/cpu\n",
+        "!pip install scikit-build cmake ninja\n",
         "!pip install git+https://github.com/Tim-Salzmann/l4casadi --no-build-isolation"
       ]
     },

diff --git a/examples/nerf_trajectory_optimization/NeRF_Trajectory_Optimization.ipynb b/examples/nerf_trajectory_optimization/NeRF_Trajectory_Optimization.ipynb
@@ -84,7 +84,8 @@
       "outputs": [],
       "source": [
         "# @title\n",
-        "!pip install torch scikit-build cmake ninja\n",
+        "!pip install torch --index-url https://download.pytorch.org/whl/cpu\n",
+        "!pip install scikit-build cmake ninja\n",
         "!pip install git+https://github.com/Tim-Salzmann/l4casadi --no-build-isolation"
       ]
     },

diff --git a/libl4casadi/CMakeLists.txt b/libl4casadi/CMakeLists.txt
@@ -1,11 +1,9 @@
 cmake_minimum_required(VERSION 3.0 FATAL_ERROR)
 project(L4CasADi)
 
-#if(NOT DEFINED ENV{DISABLE_CUDA})
-#    # Load CUDA if it is installed
-#    find_package(CUDAToolkit)
-#    find_package(CUDA)
-#endif ()
+# Load CUDA if it is installed
+find_package(CUDAToolkit)
+find_package(CUDA)
 
 if (WIN32)
     set (CMAKE_WINDOWS_EXPORT_ALL_SYMBOLS TRUE)