Torch (#140)

https://blog.rstudio.com/2020/09/29/torch/

http://b/169925060

Author: Philmod

gpu.Dockerfile

@@ -72,4 +72,8 @@ RUN apt-get install -y --no-install-recommends ocl-icd-opencl-dev && \
 RUN CPATH=/usr/local/cuda/targets/x86_64-linux/include install2.r --error --ncpus $ncpus --repo http://cran.rstudio.com \
     h2o4gpu
 
+# Torch: install the full package upfront otherwise it will be installed on loading the package which doesn't work for kernels
+# without internet (competitions for example). It will detect CUDA and install the proper version.
+RUN R -e 'library(torch); install_torch(reinstall = TRUE)'
+
 CMD ["R"]

package_installs.R

@@ -40,3 +40,8 @@ install_github("thomasp85/patchwork")
 install.packages("topicmodels")
 
 install.packages("tesseract")
+
+# Torch: install the full package upfront otherwise it will be installed on loading the package which doesn't work for kernels
+# without internet (competitions for example).
+library(torch)
+install_torch(reinstall = TRUE)

tests/test_torch.R

@@ -0,0 +1,18 @@
+context("torch")
+
+test_that("cpu imports", {
+  library(torch)
+  x <- array(runif(8), dim = c(2, 2, 2))
+  y <- torch_tensor(x, dtype = torch_float64())
+  expect_identical(x, as_array(y))
+})
+
+test_that("gpu imports", {
+  check_gpu()
+ 
+  library(torch)
+  x <- array(runif(8), dim = c(2, 2, 2))
+  y <- torch_tensor(x, dtype = torch_float64(), device = "cuda")
+  expect_identical(x, as_array(y$cpu()))
+})
+