nnUNet的docker

最新推荐文章于 2023-09-13 11:10:57 发布

大风的blog

最新推荐文章于 2023-09-13 11:10:57 发布

阅读量432

点赞数 1

文章标签： docker ubuntu

本文链接：https://blog.csdn.net/dafeng_xiaobai/article/details/117304124

版权

本文介绍了如何配置nnUNet的docker环境，包括从dockerhub拉取nvidia/cuda:10.2-cudnn8-devel-ubuntu18.04镜像，设置NVIDIA Container Toolkit以使用GPU，使用清华源加速pip安装，以及挂载数据和运行容器的步骤。

摘要由CSDN通过智能技术生成

nnUNet的docker

注意点：关于数据的Task21_kidneystone文件必须挂载到/root/tmp/下

1、在dockerhub上查询nvidia/cuda:10.2-cudnn8-devel-ubuntu18.04 ，版本号对应项目运行环境的需求

将其拉下来，sudo docker pull nvidia/cuda:10.2-cudnn8-devel-ubuntu18.04 ，成功后会有一个容器ID

2、查看拉下来的镜像，docker images -a 改镜像名字： docker tag 旧名字新名字

3、如果要在容器内使用显卡，需要NVIDIA Container Toolkit 则在主机按一下方式设置

参考链接:https://docs.nvidia.com/datacenter/cloud-native/container-toolkit/install-guide.html

# step1
curl https://get.docker.com | sh \
  && sudo systemctl start docker \
  && sudo systemctl enable docker
    
# step2
distribution=$(. /etc/os-release;echo $ID$VERSION_ID) \
   && curl -s -L https://nvidia.github.io/nvidia-docker/gpgkey | sudo ap