|
| 1 | +# Set home at PROJWORK in the .bashrc: |
| 2 | +```bash |
| 3 | +export HOME=/lustre/atlas/proj-shared/fus117/ |
| 4 | +``` |
| 5 | + |
| 6 | +#cd ~ |
| 7 | + |
| 8 | + |
| 9 | +#Set up CUDA: |
| 10 | +```bash |
| 11 | +wget http://developer.download.nvidia.com/compute/redist/cudnn/v5.1/cudnn-7.5-linux-x64-v5.1.tgz |
| 12 | +tar -xvf |
| 13 | +``` |
| 14 | + |
| 15 | +Add following to the submission script: |
| 16 | + |
| 17 | +```bash |
| 18 | +export LD_LIBRARY_PATH=$HOME/cuda/lib64:$LD_LIBRARY_PATH |
| 19 | +export LIBRARY_PATH=$HOME/cuda/lib64:$LIBRARY_PATH |
| 20 | +export LDFLAGS=$LDFLAGS:$HOME/cuda/lib64 |
| 21 | +export INCLUDE=$INCLUDE:$HOME/cuda/include |
| 22 | +export CPATH=$CPATH:$HOME/cuda/include |
| 23 | +export FFLAGS=$FFLAGS:$HOME/cuda/include |
| 24 | +export LOCAL_LDFLAGS=$LOCAL_LDFLAGS:$HOME/cuda/lib64 |
| 25 | +export LOCAL_INCLUDE=$LOCAL_INCLUDE:$HOME/cuda/include |
| 26 | +export LOCAL_CFLAGS=$LOCAL_CFLAGS:$HOME/cuda/include |
| 27 | +export LOCAL_FFLAGS=$LOCAL_FFLAGS:$HOME/cuda/include |
| 28 | +export LOCAL_CXXFLAGS=$LOCAL_CXXFLAGS:$HOME/cuda/include |
| 29 | +``` |
| 30 | + |
| 31 | + |
| 32 | +Add LIBRARY_PATH in addition to cudatoolkit: |
| 33 | +```bash |
| 34 | +module load cudatoolkit |
| 35 | +export LIBRARY_PATH=/opt/nvidia/cudatoolkit7.5/7.5.18-1.0502.10743.2.1/lib64:$LIBRARY_PATH |
| 36 | +``` |
| 37 | + |
| 38 | +# Download and install Anaconda |
| 39 | +wget https://repo.continuum.io/archive/Anaconda2-4.3.1-Linux-x86_64.sh |
| 40 | +sh A.. |
| 41 | + |
| 42 | + |
| 43 | +do not add PATH to .bashrc - it messes up modules for some reason |
| 44 | + |
| 45 | + |
| 46 | +# Clone the PPPL repo |
| 47 | + |
| 48 | +add ssh keys to github to ~/.ssh |
| 49 | +ssh-add ~/.ssh/olcf_github_rsa |
| 50 | + |
| 51 | +git clone git@github.com:PPPLDeepLearning/plasma-python.git |
| 52 | +cd PPPL/plasma-python |
| 53 | + |
| 54 | +Create PPPL env: |
| 55 | +conda create --name PPPL --file requirements.txt |
| 56 | + |
| 57 | +#Install mpi4py |
| 58 | + |
| 59 | +module switch PrgEnv-pgi PrgEnv-gnu |
| 60 | +export MPICC=cc |
| 61 | +python setup.py install |
| 62 | + |
| 63 | + |
| 64 | +doing custom installs with pip --user is OK |
| 65 | + |
| 66 | + |
| 67 | +#Make sure to update paths in the conf.yaml |
| 68 | + |
| 69 | + |
| 70 | +#The mass batch job submission is performed with this script: |
| 71 | +https://github.com/PPPLDeepLearning/plasma-python/blob/titan_setup/examples/prepare_pbs_configs_titan.py |
0 commit comments