OPTIX_ERROR_LIBRARY_NOT_FOUND when running docker_train.py on Windows

[WaningNight]

(base) PS C:\Users\ASUS\PycharmProjects\ever_training> python docker_train.py -s C:\Users\ASUS\Desktop\desk_undistorted -m C:\Users\ASUS\Desktop\desk_out
['docker', 'run', '--rm', '--gpus', 'all', '-v', '/tmp/NVIDIA:/tmp/NVIDIA', '-e', 'NVIDIA_DRIVER_CAPABILITIES=graphics,compute,utility', '-v', 'C:\Users\ASUS\Desktop\desk_undistorted:/data/dataset', '-v', 'C:\Users\ASUS\Desktop\desk_out:/data/output', '-p', '127.0.0.1:6009:6009', 'halfpotato/ever:latest', 'bash', '-c', 'source activate ever && python train.py -s /data/dataset -m /data/output "$@"', '_']
Running: docker run --rm --gpus all -v /tmp/NVIDIA:/tmp/NVIDIA -e NVIDIA_DRIVER_CAPABILITIES=graphics,compute,utility -v C:\Users\ASUS\Desktop\desk_undistorted:/data/dataset -v C:\Users\ASUS\Desktop\desk_out:/data/output -p 127.0.0.1:6009:6009 halfpotato/ever:latest bash -c source activate ever && python train.py -s /data/dataset -m /data/output "$@" _

==========
== CUDA ==

CUDA Version 12.2.0

Container image Copyright (c) 2016-2023, NVIDIA CORPORATION & AFFILIATES. All rights reserved.

This container image and its contents are governed by the NVIDIA Deep Learning Container License.
By pulling and using the container, you accept the terms and conditions of this license:
https://developer.nvidia.com/ngc/nvidia-deep-learning-container-license

A copy of this license is made available in this container at /NGC-DL-CONTAINER-LICENSE for your convenience.

/ever_training/ever/splinetracers/slang/optix-intrinsics.slang(509): warning 41010: non-void function does not return in all cases
uint get_payload(uint i) {
^~~~~~~~~~~

/ever_training/ever/splinetracers/slang/optix-intrinsics.slang(509): warning 41010: non-void function does not return in all cases
uint get_payload(uint i) {
^~~~~~~~~~~

Traceback (most recent call last):
File "/ever_training/train.py", line 17, in
from gaussian_renderer.ever import splinerender
File "/ever_training/gaussian_renderer/ever.py", line 20, in
from ever.splinetracers.fast_ellipsoid_splinetracer import trace_rays
File "/ever_training/ever/splinetracers/fast_ellipsoid_splinetracer.py", line 33, in
otx = sp.OptixContext(torch.device("cuda:0"))
RuntimeError: OPTIX_ERROR_LIBRARY_NOT_FOUND: Optix call 'optixInit()' failed: /ever_training/ever/splinetracers/fast_ellipsoid_splinetracer/py_binding.cpp:80)

Traceback (most recent call last):
File "C:\Users\ASUS\PycharmProjects\ever_training\docker_train.py", line 73, in
main()
File "C:\Users\ASUS\PycharmProjects\ever_training\docker_train.py", line 70, in main
subprocess.run(docker_cmd, check=True)
File "C:\ProgramData\anaconda3\Lib\subprocess.py", line 571, in run
raise CalledProcessError(retcode, process.args,
subprocess.CalledProcessError: Command '['docker', 'run', '--rm', '--gpus', 'all', '-v', '/tmp/NVIDIA:/tmp/NVIDIA', '-e', 'NVIDIA_DRIVER_CAPABILITIES=graphics,compute,utility', '-v', 'C:\Users\ASUS\Desktop\desk_undistorted:/data/dataset', '-v', 'C:\Users\ASUS\Desktop\desk_out:/data/output', '-p', '127.0.0.1:6009:6009', 'halfpotato/ever:latest', 'bash', '-c', 'source activate ever && python train.py -s /data/dataset -m /data/output "$@"', '_']' returned non-zero exit status 1.

[Obi-Nnamdi]

Same issue here. Did you find a fix?

EDIT: After searching around, I believe this issue is related to the fact that Optix is currently unsupported in WSL (would likely require a workaround to function): https://forums.developer.nvidia.com/t/optix-on-wsl-windows-subsystem-for-linux/140297/33.

[Obi-Nnamdi]

Got it working by following this guide with this CUDA 12.2 Toolkit for WSL2-Ubuntu.