Zhang Li пре 2 година
родитељ
комит
de92efaa36
3 измењених фајлова са 7 додато и 3 уклоњено
  1. 4 1
      Dockerfile
  2. 1 1
      docker-compose.yml
  3. 2 1
      environment.yml

+ 4 - 1
Dockerfile

@@ -1,5 +1,8 @@
 #FROM nvidia/cuda:10.2-runtime-ubuntu18.04 AS builder
-FROM nvidia/cuda:11.2.0-cudnn8-devel-ubuntu18.04 AS builder
+#FROM nvidia/cuda:11.2.0-cudnn8-devel-ubuntu18.04 AS builder
+#FROM nvidia/cuda:11.0-cudnn8-devel-ubuntu18.04 AS builder
+FROM ubuntu:18.04 AS builder
+
 RUN sed -i 's#archive.ubuntu.com#mirrors.aliyun.com#g' /etc/apt/sources.list  \
     && sed -i 's#security.ubuntu.com#mirrors.aliyun.com#g' /etc/apt/sources.list
 

+ 1 - 1
docker-compose.yml

@@ -4,7 +4,7 @@ services:
     hostname: idcard
     container_name: idcard
     restart: always
-    image: registry.cn-hangzhou.aliyuncs.com/sxtest/idcard:cpu
+    image: registry.cn-hangzhou.aliyuncs.com/sxtest/idcard:gpu
     privileged: true
     ipc: host
     tty: true

+ 2 - 1
environment.yml

@@ -21,6 +21,7 @@ dependencies:
       - requests
       - cpca
       - -i https://mirror.baidu.com/pypi/simple
-      - paddlepaddle-gpu==2.3.0.post112
+      - paddlepaddle
+#      - paddlepaddle-gpu==2.3.0.post110
       - -f https://www.paddlepaddle.org.cn/whl/linux/mkl/avx/stable.html
 prefix: /opt/conda/envs/py38