Conda#

Tip

We recommend trying Mamba (https://mamba.readthedocs.io/en/latest/). It is a drop in replacement for Conda and is typically considerably faster.

Conda (https://docs.conda.io/en/latest/) is a package management system.

Loading Conda#

System Installation#

We have a base install of Conda available as a module that can be accessed using

module load conda

User Installation#

You can also install your own versions of Conda in your $WORK or $HOME directory. We recommend

The first time you run Conda, you will need to initialize it. This creates some shell functions in your profile to make it easier to call Conda.

conda init $(echo $SHELL | cut -c 6-)

After doing this, you may need to log out and log back in to see the effects. In most cases, you can source your shell profile to avoid having to log out. For most users, this is source ~/.bashrc.

We additionally recommend that you disable Conda’s auto-activate base functionallity. By default, Conda will load a base environment, which can cause issues with system dependencies. In particular, applications on https://hpc.m3.smu.edu often behave in unexpected ways becuase it tries to use a Conda package instead of the correct system package. The next two commands tell Conda to prefer to save packages and environments in your $WORK directory so they don’t take up space in your $HOME.

conda config --set auto_activate_base false
conda config --prepend envs_dirs $WORK/.conda/envs
conda config --prepend pkgs_dirs $WORK/.conda/pkgs

Creating Virtual Environments from the Command Line#

For simple environments with a small number of packages, you can create an environment named conda_env (or any name of your choosing) in your $WORK directory with

conda create -n conda_env python=3.9 package1 package2 package3

The -n tells Conda what to name the environment. Here, we request Python version 3.9 and the packages package1 package2 package3 which are the packages you’d like to install (e.g. numpy, tensorflow, pandas, etc.). In general, it is a good idea install all the packages at the same time becasue Conda will do a better job of resolving dependencies.

Creating Virtual Environments From environment.yml File#

For environments that contain more than a few packages, we suggest creating a environment.yml file (note, you can name the file anything you’d like, but it is common practice to call it environment.yml.)

The basic structure of the environment.yml is:

name: conda_env
channels:
  - conda-forge
  - defaults
dependencies:
  - python>=3.9
  - package1
  - package2
  - package3
  - pip
  - pip:
    - pip_package1
    - -r requirements.txt

The name field is what the created environment will be called (it can be anything you like, but we again use the name conda_env for the example).

The next section is channels, which are the repositories where Conda will look for the requested packages. Conda prioritizes the channels from the top down, so in this case Conda will prefer the package in conda-forge over the package in defaults (typically the packages in the conda-forge are more up to date.)

The next section is dependencies and this is where you should list all of the packages you would like to install. If you have packages that need to be installed with pip, you should include pip in the dependencies as above and you can list the specific packages like the above as pip_package1, etc. and/or you can have all the pip packages in a requirments.txt file.

Once you have made the environment.yml file, you can create the environment with:

conda env create -f environment.yml -n conda_env

Examples#

The following are examples of how you might start to build your environment for a few different usage cases. These are the source files we use on the web portal, if you choose to build a custom environment from the form.

It is likely that these will not meet the needs of any specific use case, but you can add or remove packages as needed for your particular needs.

Using the HPC Portal#

If you are running interactive sessions through the portal using JupyterLab, you need to have JupyterLab installed in your environment. If it is not, the portal will not allow that environment to be used.

Your Conda environment should appear in the drop down list of Python Environments. If it is greyed out, that means that you need to install JupyterLab in the environment.

Using a base#

module load conda
conda activate conda_env

to the Custom environment settings field on the portal. It should look like:

Interactively from the terminal#

If you are running programs interactively from the terminal (e.g. using srun) just activate the virtual environment with

module load conda
conda activate conda_env

in the terminal before running any commands.

Using SBatch scripts#

If you are running programs using SBatch scripts, you should include the activation command in your script:

module load conda
conda activate conda_env

Tips and reproducibility#

  • In general, you should not update packages inside a Conda environment. Instead, you should make a new environment with the versions of the packages need and verify this works before removing any old environments that are not longer needed. This is especially true if you used pip to install anything.

  • It is a good idea to include version numbers of the packages you want (if you know them). For example, in the above, we requested Python version 3.9 or newer. Being more specific can help speed up how long it takes to set up the environment because it will reduce the number of package version Conda will consider.

  • It is best to install all of the packages when you create the environment, if possible. Conda will do a better job of resolving dependencies.

  • Conda can take a long time to resolve dependencies, see this blog post for more tips to speed up the process. Alternatively you can try using Mamba instead of Conda.

Additional Resources#