Sign inSign up

joeatodd/onednn-cuda

By joeatodd

•Updated almost 5 years ago

A oneDNN/oneMKL build with CUDA support

Image
0

120

joeatodd/onednn-cuda repository overview

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.

⁠Getting Started

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.

Tag summary

Content type

Image

Digest

Size

7.4 GB

Last updated

almost 5 years ago

docker pull joeatodd/onednn-cuda