search

naruya / dl_remote

Docker for everyday deep learning research on a remote server. (Tensorflow & Pytorch / Jax + VNC)
https://hub.docker.com/r/naruya/dl_remote
23 stars 6 forks source link

Support for plenoctrees rendering #9

Open naruya opened 2 years ago

naruya commented 2 years ago

Dockerfile: https://github.com/naruya/dl_remote/blob/jaxnerf/Dockerfile docker pull naruya/dl_remote:jaxnerf

$ ssh foo -L 8888:localhost:18888 -L 6006:localhost:16006 -L 5900:localhost:15900
$ docker run --gpus 0 -it -p 15900:5900 -p 16006:6006 -p 18888:8888 -v /path/to/share:/root/workspace --name name_jaxnerf naruya/dl_remote:jaxnerf
% pip install torch==1.10.1+cu113 torchvision==0.11.2+cu113 torchaudio==0.10.1+cu113 -f https://download.pytorch.org/whl/cu113/torch_stable.html
% pip install opencv-python Pillow imageio imageio-ffmpeg ipdb pyyaml pymcubes matplotlib moviepy tqdm svox
% pip install torch==1.10.1+cu113 torchvision==0.11.2+cu113 torchaudio==0.10.1+cu113 -f https://download.pytorch.org/whl/cu113/torch_stable.html
% git clone git@github.com:naruya/svox.git && cd svox && git checkout rgba_render && MAX_JOBS=16 pip install -vvv . && cd ..
% pip install jax\[cuda11_cudnn82\] -f https://storage.googleapis.com/jax-releases/jax_releases.html
% cd /path/to/workspace
% git clone https://github.com/sxyu/plenoctree
# do plenoctrees rendering

(training is not supported now.)

naruya commented 2 years ago

https://github.com/sxyu/svox2/issues/15

naruya commented 2 years ago

svox rendering https://gist.github.com/naruya/8e667d18a3d61d72bc60249f5f33ae81

naruya commented 2 years ago

Dockerfile: https://github.com/naruya/dl_remote/blob/jaxnerf/Dockerfile docker pull naruya/dl_remote:jaxnerf

$ ssh foo -L 8888:localhost:18888 -L 6006:localhost:16006 -L 5900:localhost:15900
$ docker run --gpus 0 -it -p 15900:5900 -p 16006:6006 -p 18888:8888 -v /path/to/share:/root/workspace --name name_jaxnerf naruya/dl_remote:jaxnerf
% pip install torch==1.10.1+cu113 torchvision==0.11.2+cu113 torchaudio==0.10.1+cu113 -f https://download.pytorch.org/whl/cu113/torch_stable.html
% pip install opencv-python Pillow imageio imageio-ffmpeg ipdb pyyaml pymcubes matplotlib moviepy tqdm svox
% pip install torch==1.10.1+cu113 torchvision==0.11.2+cu113 torchaudio==0.10.1+cu113 -f https://download.pytorch.org/whl/cu113/torch_stable.html
% git clone git@github.com:naruya/svox.git && cd svox && git checkout rgba_render && MAX_JOBS=16 pip install -vvv . && cd ..
% pip install jax\[cuda11_cudnn82\] -f https://storage.googleapis.com/jax-releases/jax_releases.html
% cd /path/to/workspace
% git clone https://github.com/sxyu/plenoctree
# do plenoctrees rendering

(training is not supported now.)