Ous Abou Ras

Results 5 comments of Ous Abou Ras

I am getting this error when after uncommenting and running the model. For context, I don't get any memory issues with TensorVM model. Any ideas why? ``` RuntimeError: CUDA out...

Commented back n_lamb_sigma and n_lamb_sh and its working again! not with the same great results though. I guess they need to be lower for a 12gb gpu.

fixed: just had to install the latests pytorch but wrapped in CUDA by doing ` conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch ` might be worth updating the instructions...

`echo %CUDA_HOME%` returns `C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.6` `nvcc --version` returns `Cuda compilation tools, release 11.6, V11.6.55 Build cuda_11.6.r11.6/compiler.30794723_0`

Yes pytorch is working with cuda. I have cuda 11.6 and cub 1.15.0 (compaitable with cuda 11.6) installed. I tried the recommended cuda and cub versions previously, but it would...