Skip to content
Snippets Groups Projects
Unverified Commit fe0eb0a5 authored by yancong's avatar yancong Committed by GitHub
Browse files

fix(CircleCI): Set MKL_THREADING_LAYER=GNU in dockefile (#484)

parent 5a7bc5c3
No related branches found
No related tags found
No related merge requests found
......@@ -4,6 +4,10 @@ ARG CUDNN="7"
FROM pytorch/pytorch:${PYTORCH}-cuda${CUDA}-cudnn${CUDNN}-devel
# Set MKL_THREADING_LAYER=GNU to fix issue:
# https://github.com/pytorch/pytorch/issues/37377
ENV MKL_THREADING_LAYER GNU
# To fix GPG key error when running apt-get update
RUN apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/3bf863cc.pub
RUN apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/machine-learning/repos/ubuntu1804/x86_64/7fa2af80.pub
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment