Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

[ENH] - Tensorflow with GPUs support #1890

Closed
a-babarytskyi opened this issue Mar 13, 2023 · 3 comments
Closed

[ENH] - Tensorflow with GPUs support #1890

a-babarytskyi opened this issue Mar 13, 2023 · 3 comments
Labels
type:Enhancement A proposed enhancement to the docker images

Comments

@a-babarytskyi
Copy link

What docker image(s) is this feature applicable to?

tensorflow-notebook

What changes are you proposing?

I propose to add GPUs support, because currently when I try to launch GPU calculations I get an error, I try to install inline Conda packages as described on tensor flow webpage, but it doesn't solve the problem.

How does this affect the user?

  • Users will be able to use GPUs
  • Users will waste less time on trying to configure GPUs to work with TF

Anything else?

No response

@a-babarytskyi a-babarytskyi added the type:Enhancement A proposed enhancement to the docker images label Mar 13, 2023
@mathbunnyru
Copy link
Member

@GoldenCorgo is your issue the same as described here?

@mathbunnyru
Copy link
Member

Closing as duplicate.

@mathbunnyru mathbunnyru closed this as not planned Won't fix, can't repro, duplicate, stale Mar 22, 2023
@benz0li
Copy link
Contributor

benz0li commented Mar 25, 2023

@GoldenCorgo You may be interested in the CUDA-enabled JupyterLab Python docker stack, then:

[...]

Prerequisites

The same as the JupyterLab Python docker stack plus

  • NVIDIA GPU
  • NVIDIA Linux driver
  • NVIDIA Container Toolkit

ℹ️ The host running the GPU accelerated images only requires the NVIDIA driver, the CUDA toolkit does not have to be installed.

[...]

Similar projects

What makes this project different:

  1. Multi-arch: linux/amd64, linux/arm64/v8
  2. Derived from nvidia/cuda:11.8.0-cudnn8-devel-ubuntu22.04
    • including development libraries and headers
  3. TensortRT and TensorRT plugin libraries
    • including development libraries and headers
  4. IDE: code-server next to JupyterLab
  5. Just Python – no Conda / Mamba

See Notes for tweaks, settings, etc.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
type:Enhancement A proposed enhancement to the docker images
Projects
None yet
Development

No branches or pull requests

3 participants