A oneDNN/oneMKL build with CUDA support
120
A docker image for oneDNN with CUDA support. The build script for oneDNN and dependencies is based on build_minimal.sh from https://github.com/Michoumichmich/oneAPI-setup-script with some modifications.
Get the image from https://hub.docker.com/r/joeatodd/onednn-cuda.
Launch a container with:
docker run --gpus all -it joeatodd/onednn-cuda:latest
Tools are installed in /root/sycl_workspace/deploy. PATH and LD_LIBRARY_PATH already point to the relevant subdirectories, and CC and CXX point to DPC++'s clang & clang++ respectively.
Content type
Image
Digest
Size
7.4 GB
Last updated
almost 5 years ago
docker pull joeatodd/onednn-cuda