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

前端 未结 1 2019
借酒劲吻你
借酒劲吻你 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)
提交回复
热议问题