Skip to content

perlmutter.rst

Latest commit

 

History

History
64 lines (40 loc) · 1.52 KB

perlmutter.rst

File metadata and controls

64 lines (40 loc) · 1.52 KB

Perlmutter (NERSC)

One-time Setup

To install SmartSim on Perlmutter, follow these steps:

Step 1: Create and activate a conda environment for SmartSim:

module load conda cudatoolkit/12.2 cudnn/8.9.3_cuda12 PrgEnv-gnu
conda create -n smartsim python=3.11
conda activate smartsim

Step 2: Build the SmartRedis C++ and Fortran libraries:

git clone https://github.com/CrayLabs/SmartRedis.git
cd SmartRedis
make lib-with-fortran
pip install .
cd ..

Step 3: Install SmartSim in the conda environment:

pip install git+https://github.com/CrayLabs/SmartSim.git

Step 4: Build Redis, RedisAI, the backends, and all the Python packages:

smart build --device=cuda-12

Step 5: Check that SmartSim has been installed and built correctly:

smart validate --device gpu

The following output indicates a successful install:

[SmartSim] INFO Verifying Tensor Transfer
[SmartSim] INFO Verifying Torch Backend
[SmartSim] INFO Verifying ONNX Backend
[SmartSim] INFO Verifying TensorFlow Backend
16:26:35 login SmartSim[557020:MainThread] INFO Success!

Post-installation

After completing the above steps to install SmartSim in a conda environment, you can reload the conda environment by running the following commands:

module load conda cudatoolkit/12.2 cudnn/8.9.3_cuda12 PrgEnv-gnu
conda activate smartsim