当前位置:   article > 正文

ubuntu nvidia-docker 安装教程(亲测有效)_ubuntu安装nvidia-docker

ubuntu安装nvidia-docker

1. 目的


为了能在容器里调用GPU

2.过程

  1. # If you have nvidia-docker 1.0 installed: we need to remove it and all existing GPU containers
  2. docker volume ls -q -f driver=nvidia-docker | xargs -r -I{} -n1 docker ps -q -a -f volume={} | xargs -r docker rm -f
  3. sudo apt-get purge -y nvidia-docker
  4. # Add the package repositories
  5. curl -s -L https://nvidia.github.io/nvidia-docker/gpgkey | \
  6. sudo apt-key add -
  7. distribution=$(. /etc/os-release;echo $ID$VERSION_ID)
  8. curl -s -L https://nvidia.github.io/nvidia-docker/$distribution/nvidia-docker.list | \
  9. sudo tee /etc/apt/sources.list.d/nvidia-docker.list
  10. sudo apt-get update
  11. # Install nvidia-docker2 and reload the Docker daemon configuration
  12. sudo apt-get install -y nvidia-docker2
  13. sudo pkill -SIGHUP dockerd
  14. # Test nvidia-smi with the latest official CUDA image
  15. docker run --runtime=nvidia --rm nvidia/cuda:9.0-base nvidia-smi

参考:安装NVIDIA-DOCKER - 简书

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/article/detail/47094
推荐阅读
相关标签
  

闽ICP备14008679号