pycharm+Docker+GPU配置

PyCharm Docker GPU配置指南

原创于 2019-10-15 14:06:45 发布 · 938 阅读

3 ·

CC 4.0 BY-SA版权

文章标签：

#docker # pycharm #gpu

工具同时被 2 个专栏收录

7 篇文章

订阅专栏

系统

3 篇文章

订阅专栏

本文详细介绍了如何在PyCharm专业版中配置Docker环境，并特别关注GPU的设置，以便于在Docker容器内运行和调试深度学习程序。通过修改daemon.json文件启用NVIDIA运行时，确保CUDA和cuDNN支持。

部署运行你感兴趣的模型镜像

要求

pycharm为专业版；
已经安装好docker、nvidia-docker2，此部分安装可参见博客deepin安装docker；
cuda、cudnn安装好

配置步骤：

打开pycharm，点击File->Settings->Project Interpreter

在这里插入图片描述
2. 点击add…，出现界面如下：

3. 点击界面左侧的 Docker
右侧界面Server: 点击New, 配置如下：

Image name选择自己需要的镜像，最终配置界面如下

此时，点击OK已经可以调试CPU版的程序了。

配置GPU

sudo gedit /etc/docker/daemon.json
添加"default-runtime": "nvidia"如下：

{
    "default-runtime": "nvidia",
    "runtimes": {
        "nvidia": {
            "path": "/usr/bin/nvidia-container-runtime",
            "runtimeArgs": [ ]
        }
    }
}

重启docker

sudo systemctl daemon-reload
sudo systemctl restart docker

可以在docker镜像上愉快得运行调试程序了

您可能感兴趣的与本文相关的镜像

PyTorch 2.5

PyTorch

Cuda

PyTorch 是一个开源的 Python 机器学习库，基于 Torch 库，底层由 C++ 实现，应用于人工智能领域，如计算机视觉和自然语言处理