pub_soft/alphafold
3
0
Fork 0
mirror of https://github.com/google-deepmind/alphafold.git synced 2026-06-04 14:58:05 +08:00
Code Issues Packages Projects Releases Wiki Activity

Set environment variables before any other code is executed.

Browse Source 
PiperOrigin-RevId: 495028896
Change-Id: I3b1a6ddceca1961bffdaba443e7db47bbfbc4fad
This commit is contained in:
Andrew Cowie
2022-12-13 08:26:52 -08:00
committed by Copybara-Service
parent 569eb4fea3
commit a0b0cd9cc9
1 changed files with 6 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
notebooks/AlphaFold.ipynb
View File

@@ -65,6 +65,11 @@
},
"outputs": [],
"source": [
"# Set environment variables before running any other code.\n",
"import os\n",
"os.environ['TF_FORCE_UNIFIED_MEMORY'] = '1'\n",
"os.environ['XLA_PYTHON_CLIENT_MEM_FRACTION'] = '4.0'\n",
"\n",
"#@title 1. Install third-party software\n",
"\n",
"#@markdown Please execute this cell by pressing the _Play_ button\n",
@@ -190,12 +195,7 @@
"# Make sure everything we need is on the path.\n",
"import sys\n",
"sys.path.append('/opt/conda/lib/python3.8/site-packages')\n",
"sys.path.append('/content/alphafold')\n",
"\n",
"# Make sure all necessary environment variables are set.\n",
"import os\n",
"os.environ['TF_FORCE_UNIFIED_MEMORY'] = '1'\n",
"os.environ['XLA_PYTHON_CLIENT_MEM_FRACTION'] = '2.0'"
"sys.path.append('/content/alphafold')"
]
},
{