My first GPU is good to go, but every time I ran program on the second GPU, the following error comes out.
RuntimeError: cublas runtime error : resource allocation failed at /opt/conda/conda-bld/pytorch_1524586445097/work/aten/src/THC/THCGeneral.cpp:411
Any idea on this kind of issue? I can see both GPUs in terminal by typing nvidia-smi.