ubuntu20.04使用docker容器
使用docker起因
刚开始学习机器学习、深度学习都是使用的笔记本(只有集显);后来做NLP项目需要训练bert这种大模型,训练是必须要用GPU,于是租用了colab来训练,训练后模型下载到本地pc上用cpu做推理,google driver 15G的容量勉强能满足工作需求。
再后来学习tensorrt时,colab每次登录要重新配置环境,过于耗时,于是放弃。自己买了带GPU的台式机,在ubuntu20.04上配置了tensorrt学习环境,但期间遇到很多和ubuntu20.04不兼容的情况,重装ubuntu过于麻烦,此时有朋友推荐docker,了解了下发现,docker类似与轻量化的虚拟机,这样就可以针对不同项目定制不同的docker环境,也不用担心破坏了宿主机的环境。
第一次使用docker
1.安装docker参考
https://www.runoob.com/docker/ubuntu-docker-install.html
注意:以下内容假设已经在ubuntu20.04下安装好docker
2.使用docker创建ubuntu16.04环境
root@aiden:/home/aiden# docker pull ubuntu:16.04