如何在docker容器中使用GPU以及使用pytorch

最新推荐文章于 2025-07-15 14:29:44 发布

六十英里

最新推荐文章于 2025-07-15 14:29:44 发布

阅读量1.2w

点赞数 13

CC 4.0 BY-SA版权

分类专栏：安装文章标签： docker

本文链接：https://blog.csdn.net/li_ellin/article/details/107180516

本文介绍了如何在Docker容器中使用GPU，首先确保本机NVIDIA卡可用，安装NVIDIA Container Runtime，接着安装Docker 19.03并重启。之后测试GPU功能，然后在容器内搭建PyTorch开发环境，包括安装Python、numpy和PyTorch。此外，还分享了如何管理Docker容器，提交新镜像以及镜像的保存和加载，以实现环境的便捷复用和分享。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

最近刚接触docker，如有不当，请指出。

1. 首先要解决在docker中可以使用GPU的问题。

官方给出了解决办法：https://github.com/NVIDIA/nvidia-docker

具体操作如下:

step 1:确认本机可以检测到NVIDIA卡，通过下列命令看自己GPU卡的型号。

lspci -vv | grep -i nvidia

step 2:新建nvidia-container-runtime-script.sh文件

touch nvidia-container-runtime-script.sh
vi nvidia-container-runtime-script.sh

并写入内容如下：

sudo curl -s -L https://nvidia.github.io/nvidia-container-runtime/gpgkey | \
  sudo apt-key add -
distribution=$(. /etc/os-release;echo $ID$VERSION_ID)
sudo curl -s -L https://nvidia.github.io/nvidia-container-runtime/$distribution/nvidia-container-runtime.list | \
  sudo tee /etc/apt/sources.list.d/nvidia-container-runtime.list
sudo apt-get update

执行下面语句，完成NVIDIA Container Runtime的安装：

sh nvidia-container-runtime-script.sh
apt-get install nvidia-container-run

最低0.47元/天解锁文章

200万优质内容无限畅学