|
|
@@ -10,12 +10,15 @@
|
|
|
**若您通过 Docker 安装**,请参考下述命令,使用飞桨官方 Docker 镜像,创建一个名为 `paddlex` 的容器,并将当前工作目录映射到容器内的 `/paddle` 目录:
|
|
|
|
|
|
```bash
|
|
|
-# 对于 gpu 用户
|
|
|
+# 对于 cpu 用户:
|
|
|
+nvidia-docker run --name paddlex -v $PWD:/paddle --shm-size=8G --network=host -it registry.baidubce.com/paddlepaddle/paddle:3.0.0b1 /bin/bash
|
|
|
+
|
|
|
+# 对于 gpu 用户:
|
|
|
# CUDA11.8 用户
|
|
|
nvidia-docker run --name paddlex -v $PWD:/paddle --shm-size=8G --network=host -it registry.baidubce.com/paddlepaddle/paddle:3.0.0b1-gpu-cuda11.8-cudnn8.6-trt8.5 /bin/bash
|
|
|
|
|
|
# CUDA12.3 用户
|
|
|
-nvidia-docker run --name paddlex -v $PWD:/paddle --shm-size=8G --network=host -it registry.baidubce.com/paddlepaddle/paddle:3.0.0b1-gpu-cuda12.3-cudnn9.0-trt8.6 /bin/bash
|
|
|
+nvidia-docker run --name paddlex -v $PWD:/paddle --shm-size=8G --network=host -it registry.baidubce.com/paddlepaddle/paddle:3.0.0b1-gpu-cuda12.3-cudnn9.0-trt8.6 /bin/bash
|
|
|
```
|
|
|
注:更多飞桨官方 docker 镜像请参考[飞桨官网](https://www.paddlepaddle.org.cn/install/quick?docurl=/documentation/docs/zh/install/docker/linux-docker.html)。若您是 CUDA11.8 用户,请确保您的 Docker版本 >= 19.03;若您是 CUDA12.3 用户,请确保您的 Docker版本 >= 20.10。
|
|
|
|