README_cn.md
September 18, 2023 · View on GitHub
仓库介绍
欢迎来到这个仓库,这里提供了一个基于 Docker 构建的深度学习框架环境。下面,您将找到有关环境和可用镜像的详细信息:
环境配置
- Python 版本:3.8
- PyTorch 版本:2.0
- CUDA 版本:12.0
- cuDNN 版本:8
WSLg 支持
如果您正在使用 WSLg 并希望在容器中启用图形界面支持,请按照以下步骤操作:
- 在开始之前,请确保您的 WSL 的
/tmp/.X11-unix目录已链接到mnt\wslg\.X11-unix。这是确保容器能够访问 X11 Unix 套接字的重要步骤。 - 启动容器时,请添加以下命令以将 X11 Unix 套接字挂载到容器中:
-v \\wsl$\Ubuntu-20.04\mnt\wslg\.X11-unix:/tmp/.X11-unix
可用镜像
1. PyTorch 2.0 环境(基于 CUDA 12.0)
这个镜像已经预装了 PyTorch 2.0,与 CUDA 12.0 兼容。请注意,由于包含了 PyTorch 2.0,镜像相对较大,大约占用 9.35GB 的磁盘空间。
Docker Pull 命令:
docker pull mortals/codeenv:pytorch2.0-cuda12.0
2. Miniconda 环境(基于 CUDA 12.0,无 PyTorch)
这个镜像包含 Miniconda 并适用于 CUDA 12.0,但不包含预装的 PyTorch。镜像大小约为 4.85GB。
Docker Pull 命令:
docker pull mortals/codeenv:conda-cuda12.0
3. Miniconda 环境(基于 CUDA 11.8)
此镜像与 CUDA 11.8 兼容,镜像大小约为 3.78GB。
Docker Pull 命令:
docker pull mortals/codeenv:conda-cuda11.8
包含的脚本
作者在这些镜像中提供了两个方便的脚本,以帮助您配置和启动服务:
sshstart.sh:用于配置 SSH 服务的自启动。在使用这个脚本之前,请确保在容器配置中提前进行了端口映射。SSH 服务端口默认设置为 22。jupyter.sh:用于启动 Jupyter 服务。同样,请在容器配置中提前进行端口映射,Jupyter 服务端口默认设置为 8888。
这些脚本可以帮助您更轻松地管理和访问容器中的服务。如果您需要更多详细信息或有任何其他问题,请随时联系我们。