pytorch201-cu117-mamba141#

PyTorch#

PyTorch is an optimized tensor library for deep learning using GPUs and CPUs.

Source Specifications#

MolSSI Container Hub Specifications#

  • Image pull command:

    docker pull molssi/pytorch201-cu117-mamba141:latest
    
  • Container run command:

    docker run -it --name pytorch -v $(pwd):/home molssi/pytorch201-cu117-mamba141:latest /bin/bash
    

Note

By default, the -v $(pwd):/home option mounts the current working directory to /home in your running container. Doing so, the contents of your current working directory become available in your running container. If you do not wish this to happen, you may simply remove or change this option. For NVIDIA GPU support with nvidia containers, add the --runtime nvidia --gpus all flags to the previous container run command and then run nvidia-smi to make sure all available GPUs on the docker host are visible inside the docker container.

Image Specifications#

  • OS/Arch: debian:bullseye-slim (linux/amd64)

  • Users (UID): root (0)

  • Groups (GID): root (0)

  • Environment variables:

    • CONDA_PREFIX: /opt/conda

    • PATH: ${CONDA_PREFIX}/bin:$PATH

  • Volumes: None

  • Network: None

  • Extras:

    • Added directories: None

    • Important packages installed: pytorch-cuda 11.7, torchaudio 2.0.2, torchvision 0.15.2