This commit is contained in:
miguelgfierro 2022-05-30 11:24:03 +02:00
Родитель 3b3806289f
Коммит df44178c3f
1 изменённых файлов: 2 добавлений и 2 удалений

Просмотреть файл

@ -4,7 +4,7 @@
ARG ENV="cpu"
ARG HOME="/root"
FROM ubuntu:18.04 AS base
FROM mcr.microsoft.com/mirror/docker/library/ubuntu:18.04 AS base
LABEL maintainer="Microsoft Recommender Project <RecoDevTeam@service.microsoft.com>"
@ -87,7 +87,7 @@ RUN if [ "${VIRTUAL_ENV}" = "conda" ] ; then pip install recommenders[spark,exam
###########
# GPU Stage
###########
FROM nvidia/cuda:11.2.2-cudnn8-runtime-ubuntu18.04 AS gpu
FROM mcr.microsoft.com/mirror/nvcr/nvidia/cuda:11.4.2-cudnn8-runtime-ubuntu18.04 AS gpu
ARG HOME
ARG VIRTUAL_ENV