Skip to contents

Function for installing the necessary python modules.

Usage

install_py_modules(
  envname = "aifeducation",
  install = "pytorch",
  transformer_version = "<=4.46",
  tokenizers_version = "<=0.20.4",
  pandas_version = "<=2.2.3",
  datasets_version = "<=3.1.0",
  codecarbon_version = "<=2.8.0",
  safetensors_version = "<=0.4.5",
  torcheval_version = "<=0.0.7",
  accelerate_version = "<=1.1.1",
  pytorch_cuda_version = "12.1",
  python_version = "3.9",
  remove_first = FALSE
)

Arguments

envname

string Name of the environment where the packages should be installed.

install

character determining which machine learning frameworks should be installed.

  • install = "all": for 'pytorch' and 'tensorflow'.

  • install = "pytorch": for 'pytorch'.

  • install = "tensorflow": for 'tensorflow'.

transformer_version

string determining the desired version of the python library 'transformers'.#'

tokenizers_version

string determining the desired version of the python library 'tokenizers'.

pandas_version

string determining the desired version of the python library 'pandas'.

datasets_version

string determining the desired version of the python library 'datasets'.

codecarbon_version

string determining the desired version of the python library 'codecarbon'.

safetensors_version

string determining the desired version of the python library 'safetensors'.

torcheval_version

string determining the desired version of the python library 'torcheval'.

accelerate_version

string determining the desired version of the python library 'accelerate'.#'

pytorch_cuda_version

string determining the desired version of 'cuda' for ' PyTorch'.

python_version

string Python version to use.

remove_first

bool If TRUE removes the environment completely before recreating the environment and installing the packages. If FALSE the packages are installed in the existing environment without any prior changes.

Value

Returns no values or objects. Function is used for installing the necessary python libraries in a conda environment.

See also

Other Installation and Configuration: check_aif_py_modules(), install_aifeducation(), set_transformers_logger()