diff --git a/Dockerfile.cpu b/Dockerfile.cpu index 506b734..33551d5 100644 --- a/Dockerfile.cpu +++ b/Dockerfile.cpu @@ -172,7 +172,7 @@ RUN pip --no-cache-dir install git+git://github.com/Theano/Theano.git@${THEANO_V echo "[global]\ndevice=cpu\nfloatX=float32\nmode=FAST_RUN \ \n[lib]\ncnmem=0.95 \ \n[nvcc]\nfastmath=True \ - \n[blas]\nldflag = -L/usr/lib/openblas-base -lopenblas \ + \n[blas]\nldflags = -L/usr/lib/ -lblas \ \n[DebugMode]\ncheck_finite=1" \ > /root/.theanorc diff --git a/Dockerfile.gpu b/Dockerfile.gpu index 3bad5a4..1fd3442 100644 --- a/Dockerfile.gpu +++ b/Dockerfile.gpu @@ -178,7 +178,7 @@ RUN pip --no-cache-dir install git+git://github.com/Theano/Theano.git@${THEANO_V echo "[global]\ndevice=gpu\nfloatX=float32\noptimizer_including=cudnn\nmode=FAST_RUN \ \n[lib]\ncnmem=0.95 \ \n[nvcc]\nfastmath=True \ - \n[blas]\nldflag = -L/usr/lib/openblas-base -lopenblas \ + \n[blas]\nldflags = -L/usr/lib/ -lblas \ \n[DebugMode]\ncheck_finite=1" \ > /root/.theanorc