Commit c63427af authored by Christopher Jan-Steffen Brix's avatar Christopher Jan-Steffen Brix
Browse files

Removed CuDNN

parent fe302b1c
Pipeline #110037 passed with stage
in 4 minutes and 30 seconds
......@@ -43,7 +43,7 @@ integrationMXNetJobLinux:
stage: linux
- mvn -B clean install --settings settings.xml -Dtest=IntegrationCaffe2Test
......@@ -51,19 +51,9 @@ RUN apt-get update && \
RUN python -c 'import lmdb' && echo "LMDB success" || echo "LMDB failure"
RUN apt remove nvidia-*
RUN apt update
RUN apt-key adv --fetch-keys
RUN bash -c 'echo "deb /" > /etc/apt/sources.list.d/cuda.list'
RUN apt update
RUN apt install -y cuda-toolkit-9.2
ENV CUDNN_TAR_FILE="cudnn-8.0-linux-x64-v6.0.tgz"
RUN tar -xzvf ${CUDNN_TAR_FILE}
RUN cp -P cuda/include/cudnn.h /usr/local/cuda/include
RUN cp -P cuda/lib64/libcudnn* /usr/local/cuda/lib64/
RUN chmod a+r /usr/local/cuda/lib64/libcudnn*
\ No newline at end of file
RUN apt install -y cuda-toolkit-9.2
\ No newline at end of file
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment