Open4

SRWarp ONNX export try

xhost +local: && \
  docker run --gpus all -it --rm \
  -v `pwd`:/home/user/workdir \
  -v /tmp/.X11-unix/:/tmp/.X11-unix:rw \
  --device /dev/video0:/dev/video0:mwr \
  --net=host \
  -e XDG_RUNTIME_DIR=$XDG_RUNTIME_DIR \
  -e DISPLAY=$DISPLAY \
  --privileged \
  ghcr.io/pinto0309/openvino2tensorflow:latest

sudo apt update
sudo apt install libxcb-xinerama0
sudo pip3 install PyQt5

pip install torch==1.8.1+cu111 torchvision==0.9.1+cu111 torchaudio==0.8.1 -f https://download.pytorch.org/whl/torch_stable.html

git clone https://github.com/sanghyun-son/pysrwarp && cd pysrwarp
nano Makefile
NVCC = /usr/local/cuda/bin/nvcc
INCFLAGS = -I /usr/local/cuda/include
CUDAFLAGS = -shared -O2 -gencode=arch=compute_86,code=sm_86 -std=c++14
sudo make clean && sudo make
cd ..
git clone https://github.com/sanghyun-son/srwarp && cd srwarp
cd pretrained
sudo chmod +x ./download.sh && ./download.sh
cd ../src
python3 interactive.py --pretrained ../pretrained/srwarp_mdsr-c66c4715.ckpt
RuntimeError: CUDA out of memory. Tried to allocate 20.00 MiB (GPU 0; 7.77 GiB total capacity; 5.63 GiB already allocated; 35.25 MiB free; 5.77 GiB reserved in total by PyTorch)
ログインするとコメントできます