fixing install issues in foldcond notebook

sokrypton · Jul 3, 2024 · 7aa8e26 · 7aa8e26
1 parent ed4b013
commit 7aa8e26
Showing 1 changed file with 7 additions and 4 deletions.
diff --git a/rf/examples/diffusion_foldcond.ipynb b/rf/examples/diffusion_foldcond.ipynb
@@ -74,9 +74,12 @@
         "\n",
         "  # install RFdiffusion\n",
         "  os.system(\"git clone https://github.com/sokrypton/RFdiffusion.git\")\n",
-        "  os.system(\"pip -q install jedi omegaconf hydra-core icecream pyrsistent\")\n",
-        "  os.system(\"pip install dgl -f https://data.dgl.ai/wheels/cu121/repo.html\")\n",
-        "  os.system(\"cd RFdiffusion/env/SE3Transformer; pip -q install --no-cache-dir -r requirements.txt; pip -q install .\")\n",
+        "  os.system(\"pip install jedi omegaconf hydra-core icecream pyrsistent pynvml decorator\")\n",
+        "  os.system(\"pip install git+https://github.com/NVIDIA/dllogger#egg=dllogger\")\n",
+        "  # 17Mar2024: adding --no-dependencies to avoid installing nvidia-cuda-* dependencies\n",
+        "  os.system(\"pip install --no-dependencies dgl==2.0.0 -f https://data.dgl.ai/wheels/cu121/repo.html\")\n",
+        "  os.system(\"pip install --no-dependencies e3nn==0.3.3 opt_einsum_fx\")\n",
+        "  os.system(\"cd RFdiffusion/env/SE3Transformer; pip install .\")\n",
         "\n",
         "  # extras\n",
         "  os.system(\"pip -q install py3Dmol pydssp\")\n",
@@ -670,4 +673,4 @@
   },
   "nbformat": 4,
   "nbformat_minor": 0
-}
+}