Skip to content

I2PC/Flexutils-Toolkit

Repository files navigation

Flexutils Toolkit

Supported Python versions GitHub Downloads (all assets, all releases) GitHub License

Flexutils

This package provides a neural network environment integrating the deep learning algorithms needed by scipion-em-flexutils plugin.

Included Networks

  • Zernike3Deep: Semi-classical neural network to analyze continuous heterogeneity with the Zernike3D basis
  • FlexSIREN: Smooth and regularized deformation field estimated with neural networks
  • ReconSIREN: Neural network for ab initio reconstruction and global angular assignment
  • HetSIREN: Neural network heterogeneous reconstruction for real space
  • FlexConsensus: Consensus neural network for conformational landscapes

Installation

The Flexutils-Toolkit relies on Conda and Pip to install it correctly.

Before installation, Conda must be set in the PATH variable so it can be detected during installation. Once this requirement is met, the package be installed by running the file install.sh after cloning this repository.

Know issues

For some GPUs and/or drivers, there exists a bug related to the initialization of the CuFFT libraries in Tensorflow:

  {{function_node __wrapped__rfft2d_device_/job:localhost/replica:0/task:0/device:gpu:0}} failed to create cufft batched plan with scratch allocator [op:rfft2d]

If the error appears, it can be solved by reinstalling the CuFFT libraries in the flexutils-tensorflow environment through conda.

To find the right library version, please, visit the following page.

An example for Cuda 11.8 is provided below:

  conda activate flexutils-tensorflow
  conda install -c "nvidia/label/cuda-11.8.0" libcufft
  conda deactivate

About

No description, website, or topics provided.

Resources

License

Stars

Watchers

Forks

Packages

No packages published

Contributors 3

  •  
  •  
  •