nvidia driver support on ubuntu in docker from host windows - 'Found no NVIDIA driver on your system' error

前端 未结 1 2017
借酒劲吻你
借酒劲吻你 2020-12-22 13:16

I have built a docker image: ubuntu20.04 + py38 + torch, various libs(llvmlite, cuda, pyyaml etc) + my flask app. The app uses torch and it needs nvidia driver inside the co

1条回答
  •  时光取名叫无心
    2020-12-22 13:24

    It looks like you can now run Docker in Ubuntu with the Windows Subsystem for Linux (WSL 2) and do GPU-through.

    This link goes through installation, setup and running a TensorFlow Jupyter notebook with GPU support: https://ubuntu.com/blog/getting-started-with-cuda-on-ubuntu-on-wsl-2

    Note - I haven't done this myself yet.

    0 讨论(0)
提交回复
热议问题