要求
- 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镜像上愉快得运行 调试程序了