yuzu/spleeter
1
0
Fork 0
mirror of https://github.com/YuzuZensai/spleeter.git synced 2026-01-06 04:32:43 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: GPU images

Browse Source
This commit is contained in:
Félix Voituret
2019-11-22 12:24:50 +01:00
parent dd7e594a4b
commit c6bba68308
4 changed files with 73 additions and 46 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
docker/conda-gpu.dockerfile
View File

@@ -1,4 +1,4 @@
FROM nvidia/cuda:10.1-cudnn7-runtime-ubuntu18.04
FROM nvidia/cuda:9.0-cudnn7-runtime-ubuntu16.04
RUN apt-get update --fix-missing \
&& apt-get install -y wget bzip2 ca-certificates curl git \
@@ -13,8 +13,7 @@ RUN apt-get update --fix-missing \
&& echo "conda activate base" >> ~/.bashrc \
&& ln -s /opt/conda/bin/conda /usr/bin/conda
RUN conda install -y cudatoolkit=9.0 \
&& conda install -y tensorflow-gpu==1.14.0 \
RUN conda install -y tensorflow-gpu==1.14.0 \
&& conda install -y -c conda-forge musdb
# RUN conda install -y -c conda-forge museval
# Note: switch to spleeter GPU once published.