Skip to content

Tensorflow custom env #46

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

Closed
robmarkcole opened this issue Jan 10, 2022 · 5 comments
Closed

Tensorflow custom env #46

robmarkcole opened this issue Jan 10, 2022 · 5 comments
Labels
enhancement New feature or request

Comments

@robmarkcole
Copy link

Request Tensorflow custom env example yml

@robmarkcole robmarkcole added the enhancement New feature or request label Jan 10, 2022
@robmarkcole
Copy link
Author

robmarkcole commented Jan 10, 2022

This recipe appears to work fine, although I haven't tested the GPU, or newer versions of tensorflow:

name: tf-gpu
channels:
  - conda-forge
  - defaults
dependencies:
  - python=3.9
  - pip
  - pip:
    - ipywidgets
  - conda
  - conda:
    - cudatoolkit=10.1
    - cudnn=7.6
    - nccl=2.4
    - pip=20.0
    - tensorflow-gpu=2.2
    - ipykernel

And for CPU:

name: tf-cpu
channels:
  - conda-forge
  - defaults
dependencies:
  - python=3.9
  - pip
  - pip:
    - ipywidgets
  - conda
  - conda:
    - tensorflow=2.6.2
    - ipykernel

@EmilyWebber
Copy link
Contributor

Hi Rob - we'll take a look at this. Out of curiosity, did you have any TensorFlow projects you were looking at starting right away?

@robmarkcole
Copy link
Author

Hi Emily, we are currently working on one for the competition :)

@EmilyWebber
Copy link
Contributor

Hi, we have a PR for this, but are working on a full notebook to demonstrate the capabilities, so I'm closing this for now.

#49

@robmarkcole
Copy link
Author

@EmilyWebber you have referenced a PR which doesn't include tensorflow from what I can see here

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
enhancement New feature or request
Projects
None yet
Development

No branches or pull requests

2 participants