-
Notifications
You must be signed in to change notification settings - Fork 0
Home
Will Whitney edited this page Oct 26, 2016
·
7 revisions
Load necessary modules (probably just add these lines to your ~/.bashrc):
module load gcc-4.9
module load OpenBLAS
module load fbcunn
module load cuda-7.0
export CC=gcc
export CXX=g++
SSH into a GPU machine, then clone torch and install it:
ssh rose2 # for example
git clone https://github.com/torch/distro.git ~/torch --recursive
cd ~/torch && ./install.sh
Important: If you have already run ./install.sh, you need to run ./clean.sh before you try installing again. Otherwise the install script will still be using whichever compilers and stuff were in your path the first time you tried to install.
Make sure you're in a machine that supports CUDA before you try to load cunn or cutorch! When you run nvidia-smi, you should see something like this:
Wed Oct 26 12:54:55 2016
+------------------------------------------------------+
| NVIDIA-SMI 352.79 Driver Version: 352.79 |
|-------------------------------+----------------------+----------------------+
| GPU Name Persistence-M| Bus-Id Disp.A | Volatile Uncorr. ECC |
| Fan Temp Perf Pwr:Usage/Cap| Memory-Usage | GPU-Util Compute M. |
|===============================+======================+======================|
| 0 GeForce GTX TITAN Off | 0000:02:00.0 Off | N/A |
| 30% 32C P8 12W / 250W | 15MiB / 6143MiB | 0% Default |
+-------------------------------+----------------------+----------------------+
| 1 GeForce GTX TITAN Off | 0000:03:00.0 Off | N/A |
| 30% 31C P8 12W / 250W | 15MiB / 6143MiB | 0% Default |
+-------------------------------+----------------------+----------------------+
+-----------------------------------------------------------------------------+
| Processes: GPU Memory |
| GPU PID Type Process name Usage |
|=============================================================================|
| No running processes found |
+-----------------------------------------------------------------------------+
If instead you see something like this, use a different machine:
Wed Oct 26 12:54:42 2016
+------------------------------------------------------+
| NVIDIA-SMI 352.79 Driver Version: 352.79 |
|-------------------------------+----------------------+----------------------+
| GPU Name Persistence-M| Bus-Id Disp.A | Volatile Uncorr. ECC |
| Fan Temp Perf Pwr:Usage/Cap| Memory-Usage | GPU-Util Compute M. |
|===============================+======================+======================|
| 0 GeForce GTX 580 Off | 0000:02:00.0 N/A | N/A |
| 30% 31C P12 N/A / N/A | 7MiB / 3071MiB | N/A Default |
+-------------------------------+----------------------+----------------------+
| 1 GeForce GTX 580 Off | 0000:03:00.0 N/A | N/A |
| 30% 36C P12 N/A / N/A | 7MiB / 3071MiB | N/A Default |
+-------------------------------+----------------------+----------------------+
+-----------------------------------------------------------------------------+
| Processes: GPU Memory |
| GPU PID Type Process name Usage |
|=============================================================================|
| 0 Not Supported |
| 1 Not Supported |
+-----------------------------------------------------------------------------+