前言
容器配置需要基于安装好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

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

被折叠的 条评论
为什么被折叠?



