Skip to content

Add GPU support to mxnet #303

@rosbo

Description

@rosbo

Because the kaggle gpu docker image has only the mxnet cpu version installed (it installs mxnet via pip install mxnet vs pip install pip install mxnet-cu90). If you want to fix that update the Dockerfile (https://github.com/Kaggle/docker-python/blob/cuda/Dockerfile) and submit a pull request.
-- https://www.kaggle.com/product-feedback/56045#latest-327622

Metadata

Metadata

Assignees

Labels

Type

No type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions