pytorch-vdsr icon indicating copy to clipboard operation
pytorch-vdsr copied to clipboard

run test.py has error on gpu?

Open RyanCV opened this issue 6 years ago • 1 comments

I run test.py, and got the following error

RuntimeError: cuda runtime error (46) : all CUDA-capable devices are busy or unavailable at /pytorch/torch/lib/THC/THCStream.cpp:40

could u tell me how to solve it? thx.

RyanCV avatar Dec 18 '17 17:12 RyanCV

@RyanCV It looks like a gpu problem, please put CUDA_VISIBLE_DEVICES=($Your_available_gpu_device_index) before python test.py and have a try.

twtygqyy avatar Dec 19 '17 16:12 twtygqyy