pytorch-vdsr
pytorch-vdsr copied to clipboard
run test.py has error on gpu?
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 It looks like a gpu problem, please put CUDA_VISIBLE_DEVICES=($Your_available_gpu_device_index) before python test.py and have a try.