apps/go-darknet
1
0
Fork 0
mirror of https://github.com/LdDl/go-darknet.git synced 2025-09-26 19:51:27 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix dockerfile.gpu

Browse Source
This commit is contained in:
lucmichalski
2020-04-08 08:02:03 +00:00
parent 3166cdca78
commit 766ff56e2e
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
docker/Dockerfile.gpu
View File

@@ -1,16 +1,16 @@
# Build phase
FROM nvidia/cuda:9.0-cudnn7-devel-ubuntu16.04 as builder
FROM nvidia/cuda:10.0-cudnn7-devel-ubuntu18.04 as builder
ENV darknet_commit=680d3bd
WORKDIR /root/build
COPY darknet.patch .
COPY Makefile.gpu .
RUN apt-get -y update && \
apt-get -y install git build-essential && \
git clone https://github.com/pjreddie/darknet.git && \
cd darknet && \
git checkout $darknet_commit && \
git apply ../darknet.patch && \
cp -f /root/build/Makefile.gpu Makefile && \
make
# Final Image