Closed yongjer closed 1 year ago
I've tried to use Katrain in docker env to use tensorrt version, and I've tested that this will work, but hope that it can be more light weighted and used in multiple host env. e.g. driver version, trt version, etc.
make sure that use "xhost +" to activate gui
example docker run command on my RTX 2060:
docker run --gpus all \ -it \ --ipc=host \ --ulimit memlock=-1 \ --ulimit stack=67108864 \ --net=host \ -v /tmp/.X11-unix:/tmp/.X11-unix \ -e DISPLAY=$DISPLAY \ katrain:latest
I've tried to use Katrain in docker env to use tensorrt version, and I've tested that this will work, but hope that it can be more light weighted and used in multiple host env. e.g. driver version, trt version, etc.
make sure that use "xhost +" to activate gui
example docker run command on my RTX 2060:
docker run --gpus all \ -it \ --ipc=host \ --ulimit memlock=-1 \ --ulimit stack=67108864 \ --net=host \ -v /tmp/.X11-unix:/tmp/.X11-unix \ -e DISPLAY=$DISPLAY \ katrain:latest