pycharm+Docker+GPU配置

要求

  1. pycharm为专业版;
  2. 已经安装好docker、nvidia-docker2,此部分安装可参见博客deepin安装docker;
  3. cuda、cudnn安装好

配置步骤:

  1. 打开pycharm, 点击File->Settings->Project Interpreter

pycharm+Docker+GPU配置_第1张图片
2. 点击add…,出现界面如下:
pycharm+Docker+GPU配置_第2张图片
3. 点击界面左侧的 Docker
右侧界面Server: 点击New, 配置如下:
pycharm+Docker+GPU配置_第3张图片
Image name选择自己需要的镜像,最终配置界面如下
pycharm+Docker+GPU配置_第4张图片
此时,点击OK已经可以调试CPU版的程序了。

配置GPU

  1. sudo gedit /etc/docker/daemon.json
    添加"default-runtime": "nvidia"如下:
{
    "default-runtime": "nvidia",
    "runtimes": {
        "nvidia": {
            "path": "/usr/bin/nvidia-container-runtime",
            "runtimeArgs": [ ]
        }
    }
}
  1. 重启docker
sudo systemctl daemon-reload
sudo systemctl restart docker

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

你可能感兴趣的:(工具,系统)