配置docker-yolov5容器,并在pycharm实现ssh连接docker解释器

本文介绍了如何在Docker环境下配置Yolov5,避免使用大体积的预构建镜像,而是选择拉取nvidia/cuda镜像并手动配置。步骤包括拉取cuda镜像,设置SSH连接,通过PyCharm进行远程解释器调试,并详细说明了在服务器端的环境配置,如更换镜像源、安装必要的软件等。此外,还展示了如何在PyCharm中设置SSH Interpreter,实现代码的同步与调试。

前言

容器配置需要基于安装好docker和nvidia-docker条件下。
这边配置容器有两条路,一种是直接在docker hub拉取yolov5的镜像,但是大小有十几个G,包含不需要的一些库。另一种是拉取nvidia/cuda镜像,并自己在里面配置环境。这边主要叙述第二种。


一、直接拉取yolov5镜像

https://hub.docker.com/
进入网址,搜索yolov5
在这里插入图片描述
在这里插入图片描述
在服务器输入如下指令即可

docker pull ultralytics/yolov5

二、自己选取cuda镜像并配置环境

1.拉取cuda镜像

搜索nvidia/cuda

在这里插入图片描述
点进去在tag选择标签号
在这里插入图片描述
我是选择11.3.1版本 11.3.1-devel-ubuntu18.04
在这里插入图片描述

直接拉取

docker pull nvidia/cuda:11.3.1-devel-ubuntu18.04

评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值