search

Kaggle / docker-rstats

Kaggle R docker image
Apache License 2.0
144 stars 90 forks source link

Refactor CUDA installation #115

Closed Philmod closed 4 years ago

Philmod commented 4 years ago

Use the same as for docker-python image.

Philmod commented 4 years ago

The build fails as the test doesn't find the GPU device: InvalidArgumentCannot assign a device for operation Const: node Const (defined at /framework/ops.py:1748) was explicitly assigned to /device:GPU:0 but available devices are [ /job:localhost/replica:0/task:0/device:CPU:0, /job:localhost/replica:0/task:0/device:XLA_CPU:0, /job:localhost/replica:0/task:0/device:XLA_GPU:0 ]. Make sure the device specification refers to a valid device.