Docker + CUDA documentation #1232
Conversation
|
I've been thinking that it might be easier to use ffnvcodec inside libvmaf rather than maintain a special Dockerfile for building this. I would like to continue being cross-platform as possible, meaning in addition to Linux, we still have support for Windows, FreeBSD, etc. Thoughts? |
|
While I agree with you this is much more work that simply doing a separate Dockerfile. I am not sure when or if I can get to do this. |
gedoensmax
force-pushed
the
docker_cuda
branch
from
7cb98cb
to
52ad584
Compare
| --enable-ffnvcodec \ | ||
| --disable-stripping \ | ||
| --extra-cflags="-I/usr/local/cuda/include" \ | ||
| --extra-ldflags="-L/usr/local/cuda/lib64 -L/usr/local/cuda/lib64/stubs/" |
There was a problem hiding this comment.
Linking against stubs as during build the driver is not available.
|
@gedoensmax I will merge this now, and we can look at non-Docker alternatives in the future. Anything else you want to add before I merge? |
|
@MorkTheOrk please take a look if you think something is missing otherwise I am happy with it. |
| (add `-Denable_float=true` flag in the rare case if you want to use the floating-point feature extractors.) | ||
| Special cases: | ||
| - add `-Denable_float=true` flag in the rare case if you want to use the floating-point feature extractors. | ||
| - add `-Denable_avx512=true` to support wider SIMD instructions to achieve the fastest processing on supported CPUs |
There was a problem hiding this comment.
nit: this is now enabled by default so this wording may be misleading. We can fix this later though.
|
@gedoensmax For the Docker and Documentation part: Looks fine to me. |
Actually the base image is a bit overkill. CUDA Toolkit is only required for compilation but not for execution. It should be sufficient to compile in a CUDA docker but copy everything over to the respective Ubuntu base image. Any help with reducing the image into a build image and a final image would be much appreciated ! If image size is not a concern this is working as is.