also install torch into the system library

mlverse · May 24, 2023 · c47d830 · c47d830
1 parent 2e08ee9
commit c47d830
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/torch/Dockerfile b/torch/Dockerfile
@@ -12,6 +12,6 @@ ARG TORCH_VERSION="v0.10.0"
 
 RUN export V="${TORCH_VERSION}" && \
     export TORCH_INSTALL=1 && \
-    Rscript -e "v <- Sys.getenv('V'); if (v == 'cran') pak::pkg_install('torch') else pak::pkg_install(sprintf('mlverse/torch@%s', v));" && \
+    Rscript -e "v <- Sys.getenv('V'); if (v == 'cran') pak::pkg_install('torch', lib=.Library) else pak::pkg_install(sprintf('mlverse/torch@%s', v), lib=.Library);" && \
     Rscript -e "pak::pak_cleanup(force=TRUE, metadata_cache=FALSE)" && \
     rm -rf /tmp/*