affnet icon indicating copy to clipboard operation
affnet copied to clipboard

Published 20 hours ago • verified publisher icon ducha-aiki

Can not run on CUDA

Open vyunolbek opened this issue 4 months ago • 2 comments

/home/old-ufo/anaconda3/envs/fastai1/lib/python3.7/site-packages/ipykernel_launcher.py(16): <module> /home/old-ufo/anaconda3/envs/fastai1/lib/python3.7/runpy.py(85): _run_code /home/old-ufo/anaconda3/envs/fastai1/lib/python3.7/runpy.py(193): _run_module_as_main RuntimeError: Expected all tensors to be on the same device, but found at least two devices, cuda:0 and cpu! (when checking argument for argument running_mean in method wrapper__native_batch_norm)

vyunolbek avatar Oct 16 '24 10:10 vyunolbek