User Tools

Site Tools


eman2:install:condainstall

Differences

This shows you the differences between two versions of the page.

Link to this comparison view

Both sides previous revisionPrevious revision
eman2:install:condainstall [2025/06/19 21:41] steveludtkeeman2:install:condainstall [2025/07/06 18:34] (current) steveludtke
Line 54: Line 54:
 </code> </code>
  
-<<Anchor(GPU)>> 
 ===== GPU Aceleration & Deep Learning ===== ===== GPU Aceleration & Deep Learning =====
 Older EMAN2 programs make use of Tensorflow for Deep Learning/GPU support, but we have been gradually migrating to the new JAX library, so having both installed is strongly suggested. Older EMAN2 programs make use of Tensorflow for Deep Learning/GPU support, but we have been gradually migrating to the new JAX library, so having both installed is strongly suggested.
Line 60: Line 59:
   * Tensorflow   * Tensorflow
     * Should be installed automatically as part of the EMAN2 environment setup above     * Should be installed automatically as part of the EMAN2 environment setup above
-    * You can check to see what version of Tensorflow and whether it is GPU accelerated with: "conda list|grep tensorflow". On a machine with an NVIDIA GPU and properly configured drivers, you should see a tensorflow version like "2.17.0    cuda120py312h02ad488_203". This says that you are using the CUDA 12.0, Python 3.12 version of tensorflow 2.17.0. If you see "cpu" instead of "cuda" then you do not have GPU acceleration, and you will need to sort that out.+    * You can check to see what version of Tensorflow and whether it is GPU accelerated with: //conda list|grep tensorflow//. On a machine with an NVIDIA GPU and properly configured drivers, you should see a tensorflow version like **2.17.0    cuda120py312h02ad488_203**. This says that you are using the CUDA 12.0, Python 3.12 version of tensorflow 2.17.0. If you see "cpu" instead of "cuda" then you do not have GPU acceleration, and you will need to sort that out.
     * You can test tensorflow with: //$CONDA_PREFIX/examples/test_tensorflow_basic.py//     * You can test tensorflow with: //$CONDA_PREFIX/examples/test_tensorflow_basic.py//
  
Line 66: Line 65:
     * While it is possible to install JAX using conda/mamba, it can be difficult to find a version of GPU JAX and a version of GPU Tensorflow which will install properly together this way.     * While it is possible to install JAX using conda/mamba, it can be difficult to find a version of GPU JAX and a version of GPU Tensorflow which will install properly together this way.
     * Instead, we strongly suggest just installing JAX using pip:     * Instead, we strongly suggest just installing JAX using pip:
-      conda activate eman2 +      <code> 
-      Linux (NVIDIA): //pip install jax[cuda12] optax//     +conda activate eman2 
-      Other (cpu): //pip install jax optax//+ 
 +Linux (NVIDIA):  
 +pip install jax[cuda12] optax    
 + 
 +#Other (cpu):  
 +pip install jax optax 
 +      </code>
     * see [[https://pypi.org/project/jax/]] for up to date JAX instructions, this may change more frequently than this page is updated     * see [[https://pypi.org/project/jax/]] for up to date JAX instructions, this may change more frequently than this page is updated
     * You can test after installation with: //$CONDA_PREFIX/examples/test_jax_basic.py//     * You can test after installation with: //$CONDA_PREFIX/examples/test_jax_basic.py//
eman2/install/condainstall.txt · Last modified: by steveludtke