PKU-EPIC/DREDS

pytorch can't use cuda

meixikyi opened this issue · 1 comments

when I run stereo_matching.py, I found pytorch can't use CUDA

File "/DepthSensorSimulator/stereo_matching.py", line 608, in force_cudnn_initialization
torch.nn.functional.conv2d(torch.zeros(s, s, s, s, device=dev), torch.zeros(s, s, s, s, device=dev))
File "/root/anaconda3/envs/DepthSensorSimulator/lib/python3.9/site-packages/torch/cuda/init.py", line 208, in _lazy_init
raise AssertionError("Torch not compiled with CUDA enabled")
AssertionError: Torch not compiled with CUDA enabled

anyone meet it before?
thanks

Daiqy commented

Hi! Looks like the PyTorch environment configuration problem. Please try to reconfigure your environment.