Skip to content
This repository has been archived by the owner on Jun 25, 2022. It is now read-only.

Commit

Permalink
nvblas linking
Browse files Browse the repository at this point in the history
  • Loading branch information
@cboettig
cboettig committed Mar 9, 2019
1 parent 85991f9 commit 0209e4f
Showing 1 changed file with 8 additions and 1 deletion.
9 changes: 8 additions & 1 deletion cuda/base/Dockerfile
View file
Original file line number Diff line number Diff line change
Expand Up @@ -73,4 +73,11 @@ RUN echo "rsession-ld-library-path=$LD_LIBRARY_PATH" | tee -a /etc/rstudio/rserv
\nCUDA_HOME=$CUDA_HOME \
\nCUDA_PATH=$CUDA_PATH \
\nPATH=$PATH" >> /usr/local/lib/R/etc/Renviron


## Allow R to use CUDA for BLAS, with fallback on openblas
ENV LD_PRELOAD=/usr/local/cuda/lib64/libnvblas.so.$CUDA_VERSION
ENV NVBLAS_CONFIG_FILE=/etc/nvblas.conf
RUN echo "NVBLAS_LOGFILE nvblas.log \
\nNVBLAS_CPU_BLAS_LIB /usr/lib/libopenblas.so
\nNVBLAS_GPU_LIST ALL" > /etc/nvblas.conf

0 comments on commit 0209e4f

Please sign in to comment.