Skip to content

Commit 598c8ba

Browse files
Notes on Titan setup that works
1 parent 82b66ef commit 598c8ba

File tree

1 file changed

+71
-0
lines changed

1 file changed

+71
-0
lines changed

docs/Titan.md

Lines changed: 71 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,71 @@
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

Comments
 (0)