在Docker上配置TensorFlow

AI小白龙*

已于 2024-05-05 09:49:31 修改

阅读量997

点赞数 23

文章标签： docker tensorflow 容器深度学习 pytorch 人工智能图像处理

于 2024-01-12 12:00:00 首次发布

本文链接：https://blog.csdn.net/2301_81940605/article/details/135532018

版权

在Docker上配置TensorFlow

配置WSL

参考教程：https://blog.csdn.net/m0_63969219/article/details/124632640

在上述教程配置的过程中，可能很难在微软商店下到ubuntu，下面给出另外一种解决方案：

接着上面教程
```
wsl --set-default-version 2
```
之后，输入
```
wsl -l -o
```
可以列出安装的系统
安装Ubuntu20.04
```
wsl --install -d Ubuntu-20.04
```

然后设置账号密码，就可以了

重启方法： （在cmd上用管理员运行）
- ```
net stop LxssManager	//停止
net start LxssManager	//启动
```
因为之后在docker中会自动启动WSL，所以主要是用到停止命令，在每次用完GPU后，记得在cmd（之前我用Windows Powershell关不了，不知道什么原因） 中 手动停止（也就是输入上面的停止命令）（最好不要直接在任务管理器关掉！）

配置docker

配置NVIDIA驱动程序

打开geforce exprience，下载安装game ready或者studio即可

下载配置docker

下载链接：https://docs.docker.com/desktop/install/windows-install/

下完之后打开docker，点击设置

点击[Resource]，点击[WSL integration]，将Ubuntu-20.04打开

然后在windows命令行下，输入

docker pull tensorflow/tensorflow:2.9.1-gpu-jupyter

开始下载

（这个版本是笔者编写时的最新版本，如果要更新，改一下版本号再pull就行了）

然后输入

docker run --gpus all -it -p 8888:8888 tensorflow/tensorflow:2.9.1-gpu-jupyter

就能再container中找到tensorflow了

可以在container中，点击[open with browser]在网页中打开

此时会发现需要输入token，点击原来那个container，可以在日志（logs）中找到token

然后就可以在网页中打开jupyter了

在此处new一个终端terminal，输入

nvidia-smi

如果有cuda版本，就说明gpu有了

接下来，就能在jupyter上进行作业了（记得每次退出docker后记得查看要不要手动关掉WSL哦）

如何学习大模型 AI ？

由于新岗位的生产效率，要优于被取代岗位的生产效率，所以实际上整个社会的生产效率是提升的。

但是具体到个人，只能说是：

“最先掌握AI的人，将会比较晚掌握AI的人有竞争优势”。

这句话，放在计算机、互联网、移动互联网的开局时期，都是一样的道理。

我在一线互联网企业工作十余年里，指导过不少同行后辈。帮助很多人得到了学习和成长。

我意识到有很多经验和知识值得分享给大家，也可以通过我们的能力和经验解答大家在人工智能学习中的很多困惑，所以在工作繁忙的情况下还是坚持各种整理和分享。但苦于知识传播途径有限，很多互联网行业朋友无法获得正确的资料得到学习提升，故此将并将重要的AI大模型资料包括AI大模型入门学习思维导图、精品AI大模型学习书籍手册、视频教程、实战学习等录播视频免费分享出来。

😝有需要的小伙伴，可以点击下方链接免费领取或者V扫描下方二维码免费领取🆓

在这里插入图片描述