Issue with GPU on CentOS7


To help you find the problem, here is a very short minimal example.

  1. First login to login node 1 (there is a problem with login node 2 I just discovered it will be covered in another message).
  2. Build the official pytorch image. For ease of use I uploaded the image on dockerhub.

singularity build pytorch.simg docker://pablostrasser/pytorch:latest

  1. Create a basic python script:

    import torch
    cuda = torch.device(‘cuda’)

  2. Execute the script:

srun -p kalousis-gpu-EL7 --gres=gpu:1 singularity exec --nv pytorch.simg python /home/strassp6/scratch/

The script fail with:

THCudaCheck FAIL file=…/aten/src/THC/THCGeneral.cpp line=51 error=999 : unknown error
Traceback (most recent call last):
File “/home/strassp6/scratch/”, line 3, in
File “/opt/conda/lib/python3.6/site-packages/torch/cuda/”, line 163, in _lazy_init
RuntimeError: cuda runtime error (999) : unknown error at …/aten/src/THC/THCGeneral.cpp:51
srun: error: gpu008: task 0: Exited with exit code 1

For reminder we had the same problem at the last update.

I hope this help.
I’m available if you have more questions.

Pablo Strasser

1 Like